MLCompilerBridge
Tools for streamlining communication with ML models for compiler optimizations.
Loading...
Searching...
No Matches
MLBridge
TensorSpec
MLBridge::TensorSpec Member List
This is the complete list of members for
MLBridge::TensorSpec
, including all inherited members.
createSpec
(const std::string &Name, const std::vector< int64_t > &Shape, int Port=6020)
MLBridge::TensorSpec
inline
static
ElementCount
MLBridge::TensorSpec
private
ElementSize
MLBridge::TensorSpec
private
getDataType
()
MLBridge::TensorSpec
private
static
getElementByteSize
() const
MLBridge::TensorSpec
inline
getElementCount
() const
MLBridge::TensorSpec
inline
getTotalTensorBufferSize
() const
MLBridge::TensorSpec
inline
isElementType
() const
MLBridge::TensorSpec
inline
Name
MLBridge::TensorSpec
private
name
() const
MLBridge::TensorSpec
inline
operator!=
(const TensorSpec &Other) const
MLBridge::TensorSpec
inline
operator==
(const TensorSpec &Other) const
MLBridge::TensorSpec
inline
Port
MLBridge::TensorSpec
private
port
() const
MLBridge::TensorSpec
inline
setShape
(const std::vector< int64_t > &NewShape)
MLBridge::TensorSpec
inline
shape
() const
MLBridge::TensorSpec
inline
Shape
MLBridge::TensorSpec
private
TensorSpec
(const std::string &NewName, const TensorSpec &Other)
MLBridge::TensorSpec
inline
TensorSpec
(const std::string &Name, int Port, TensorType Type, size_t ElementSize, const std::vector< int64_t > &Shape)
MLBridge::TensorSpec
private
toJSON
(llvm::json::OStream &OS) const
MLBridge::TensorSpec
type
() const
MLBridge::TensorSpec
inline
Type
MLBridge::TensorSpec
private
Generated by
1.10.0