Home
last modified time | relevance | path

Searched refs:input_tensors_as_shapes_ (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dshape_inference.cc177 input_tensors_as_shapes_ = input_tensors_as_shapes; in PreInputInit()
730 const int input_tensors_as_shapes_size = input_tensors_as_shapes_.size(); in MakeShapeFromShapeTensorTreatScalarAsUnknownShape()
732 input_tensors_as_shapes_[input_idx].IsSet() && in MakeShapeFromShapeTensorTreatScalarAsUnknownShape()
733 RankKnown(input_tensors_as_shapes_[input_idx])) { in MakeShapeFromShapeTensorTreatScalarAsUnknownShape()
734 *out = input_tensors_as_shapes_[input_idx]; in MakeShapeFromShapeTensorTreatScalarAsUnknownShape()
749 const int input_tensors_as_shapes_size = input_tensors_as_shapes_.size(); in MakeShapeFromShapeTensor()
751 input_tensors_as_shapes_[input_idx].IsSet() && in MakeShapeFromShapeTensor()
752 RankKnown(input_tensors_as_shapes_[input_idx])) { in MakeShapeFromShapeTensor()
753 *out = input_tensors_as_shapes_[input_idx]; in MakeShapeFromShapeTensor()
1169 const int input_tensors_as_shapes_size = input_tensors_as_shapes_.size(); in AttachContext()
[all …]
H A Dshape_inference.h384 input_tensors_as_shapes_ = input_tensors_as_shapes; in set_input_tensors_as_shapes()
388 return input_tensors_as_shapes_; in input_tensors_as_shapes()
826 std::vector<ShapeHandle> input_tensors_as_shapes_; variable