Searched defs:TensorSpec (Results 1 – 4 of 4) sorted by relevance
30 struct TensorSpec { struct34 bool operator==(const TensorSpec& o) const { argument40 friend H AbslHashValue(H h, const TensorSpec& t) { in AbslHashValue() argument
25 TensorSpec::TensorSpec() in TensorSpec() function in tensorflow::TensorSpec28 TensorSpec::TensorSpec(PartialTensorShape shape, DataType dtype) in TensorSpec() function in tensorflow::TensorSpec31 TensorSpec::TensorSpec(const TensorSpecProto& proto) in TensorSpec() function in tensorflow::TensorSpec
57 struct TensorSpec { struct61 explicit TensorSpec(tensorflow::DataType dtype) : dtype(dtype) {} in TensorSpec() argument62 TensorSpec(tensorflow::DataType dtype, tensorflow::PartialTensorShape shape) in TensorSpec() argument
118 class TensorSpec(DenseSpec, type_spec.BatchableTypeSpec): class