MLCompilerBridge
Tools for streamlining communication with ML models for compiler optimizations.
Loading...
Searching...
No Matches
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::TensorSpecinlinestatic
ElementCountMLBridge::TensorSpecprivate
ElementSizeMLBridge::TensorSpecprivate
getDataType()MLBridge::TensorSpecprivatestatic
getElementByteSize() constMLBridge::TensorSpecinline
getElementCount() constMLBridge::TensorSpecinline
getTotalTensorBufferSize() constMLBridge::TensorSpecinline
isElementType() constMLBridge::TensorSpecinline
NameMLBridge::TensorSpecprivate
name() constMLBridge::TensorSpecinline
operator!=(const TensorSpec &Other) constMLBridge::TensorSpecinline
operator==(const TensorSpec &Other) constMLBridge::TensorSpecinline
PortMLBridge::TensorSpecprivate
port() constMLBridge::TensorSpecinline
setShape(const std::vector< int64_t > &NewShape)MLBridge::TensorSpecinline
shape() constMLBridge::TensorSpecinline
ShapeMLBridge::TensorSpecprivate
TensorSpec(const std::string &NewName, const TensorSpec &Other)MLBridge::TensorSpecinline
TensorSpec(const std::string &Name, int Port, TensorType Type, size_t ElementSize, const std::vector< int64_t > &Shape)MLBridge::TensorSpecprivate
toJSON(llvm::json::OStream &OS) constMLBridge::TensorSpec
type() constMLBridge::TensorSpecinline
TypeMLBridge::TensorSpecprivate