Home
last modified time | relevance | path

Searched refs:BuildPartialTensorShape (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dtensor_shape.h541 static Status BuildPartialTensorShape(gtl::ArraySlice<int64_t> dim_sizes, in BuildPartialTensorShape() function
545 static Status BuildPartialTensorShape( in BuildPartialTensorShape() function
547 return BuildPartialTensorShape(gtl::ArraySlice<int64_t>(dim_sizes), out); in BuildPartialTensorShape()
549 static Status BuildPartialTensorShape(const TensorShapeProto& proto, in BuildPartialTensorShape() function
554 static StatusOr<PartialTensorShape> BuildPartialTensorShape( in BuildPartialTensorShape() function
H A Dresource_handle.cc91 Status s = PartialTensorShape::BuildPartialTensorShape( in FromProto()
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/utils/
H A Dshape_inference_utils.cc92 tensorflow::PartialTensorShape::BuildPartialTensorShape( in GetShapeFromMlirType()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dconstant_folding.cc1021 if (!PartialTensorShape::BuildPartialTensorShape(output_prop.shape(), in IsFoldableUncached()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes_test.cc2354 TF_CHECK_OK(PartialTensorShape::BuildPartialTensorShape( in TestConvertReadVariableOp()