Searched refs:input_tensors_as_shapes_ (Results 1 – 2 of 2) sorted by relevance
228 input_tensors_as_shapes_ = input_tensors_as_shapes; in PreInputInit()781 if (input_idx < input_tensors_as_shapes_.size() && in MakeShapeFromShapeTensorTreatScalarAsUnknownShape()782 input_tensors_as_shapes_[input_idx].IsSet() && in MakeShapeFromShapeTensorTreatScalarAsUnknownShape()783 RankKnown(input_tensors_as_shapes_[input_idx])) { in MakeShapeFromShapeTensorTreatScalarAsUnknownShape()784 *out = input_tensors_as_shapes_[input_idx]; in MakeShapeFromShapeTensorTreatScalarAsUnknownShape()799 if (input_idx < input_tensors_as_shapes_.size() && in MakeShapeFromShapeTensor()800 input_tensors_as_shapes_[input_idx].IsSet() && in MakeShapeFromShapeTensor()801 RankKnown(input_tensors_as_shapes_[input_idx])) { in MakeShapeFromShapeTensor()802 *out = input_tensors_as_shapes_[input_idx]; in MakeShapeFromShapeTensor()1164 i < input_tensors_as_shapes_.size() && in AttachContext()[all …]
319 input_tensors_as_shapes_ = input_tensors_as_shapes; in set_input_tensors_as_shapes()323 return input_tensors_as_shapes_; in input_tensors_as_shapes()755 std::vector<ShapeHandle> input_tensors_as_shapes_; variable