Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/framework/
Dshape_inference.cc170 input_tensors_as_shapes_ = input_tensors_as_shapes; in PreInputInit()
723 const int input_tensors_as_shapes_size = input_tensors_as_shapes_.size(); in MakeShapeFromShapeTensorTreatScalarAsUnknownShape()
725 input_tensors_as_shapes_[input_idx].IsSet() && in MakeShapeFromShapeTensorTreatScalarAsUnknownShape()
726 RankKnown(input_tensors_as_shapes_[input_idx])) { in MakeShapeFromShapeTensorTreatScalarAsUnknownShape()
727 *out = input_tensors_as_shapes_[input_idx]; in MakeShapeFromShapeTensorTreatScalarAsUnknownShape()
742 const int input_tensors_as_shapes_size = input_tensors_as_shapes_.size(); in MakeShapeFromShapeTensor()
744 input_tensors_as_shapes_[input_idx].IsSet() && in MakeShapeFromShapeTensor()
745 RankKnown(input_tensors_as_shapes_[input_idx])) { in MakeShapeFromShapeTensor()
746 *out = input_tensors_as_shapes_[input_idx]; in MakeShapeFromShapeTensor()
1136 const int input_tensors_as_shapes_size = input_tensors_as_shapes_.size(); in AttachContext()
[all …]
Dshape_inference.h308 input_tensors_as_shapes_ = input_tensors_as_shapes; in set_input_tensors_as_shapes()
312 return input_tensors_as_shapes_; in input_tensors_as_shapes()
740 std::vector<ShapeHandle> input_tensors_as_shapes_; variable