Searched refs:input_tensors_as_shapes (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.h | 175 const std::vector<ShapeHandle>& input_tensors_as_shapes,
191 const std::vector<PartialTensorShape>& input_tensors_as_shapes,
309 const std::vector<ShapeHandle>& input_tensors_as_shapes) { in set_input_tensors_as_shapes() argument
310 input_tensors_as_shapes_ = input_tensors_as_shapes; in set_input_tensors_as_shapes()
313 const std::vector<ShapeHandle>& input_tensors_as_shapes() const { in input_tensors_as_shapes() function
690 const std::vector<ShapeHandle>& input_tensors_as_shapes);
|
| D | shape_inference.cc | 39 const std::vector<PartialTensorShape>& input_tensors_as_shapes, in InferenceContext() argument
45 input_tensors_as_shape_handles.reserve(input_tensors_as_shapes.size()); in InferenceContext()
46 for (const PartialTensorShape& p : input_tensors_as_shapes) { in InferenceContext()
91 const std::vector<ShapeHandle>& input_tensors_as_shapes, in InferenceContext() argument
95 PreInputInit(op_def, input_tensors, input_tensors_as_shapes); in InferenceContext()
170 const std::vector<ShapeHandle>& input_tensors_as_shapes) { in PreInputInit() argument
177 input_tensors_as_shapes_ = input_tensors_as_shapes; in PreInputInit()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | shape_inference_utils.cc | 254 std::vector<ShapeHandle> input_tensors_as_shapes(num_operands); in InferReturnTypeComponentsForTFOp() local
296 !input_tensors[input] && !input_tensors_as_shapes[input].Handle()) { in InferReturnTypeComponentsForTFOp()
305 input_tensors_as_shapes[input] = handle; in InferReturnTypeComponentsForTFOp()
314 c.set_input_tensors_as_shapes(input_tensors_as_shapes); in InferReturnTypeComponentsForTFOp()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.cc | 673 std::vector<ShapeHandle> input_tensors_as_shapes; in RunShapeFn() local
678 c->set_input_tensors_as_shapes(input_tensors_as_shapes); in RunShapeFn()
762 if (i >= input_tensors_as_shapes.size()) { in RunShapeFn()
763 input_tensors_as_shapes.resize(i + 1); in RunShapeFn()
767 input_tensors_as_shapes[i] = s; in RunShapeFn()
776 c->set_input_tensors_as_shapes(input_tensors_as_shapes); in RunShapeFn()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 906 } else if (static_cast<int>(ic->input_tensors_as_shapes().size()) > i && in UpdateFunction()
908 ic, ic->input(i), ic->input_tensors_as_shapes()[i], in UpdateFunction()
913 ic, ic->input(i), ic->input_tensors_as_shapes()[i], in UpdateFunction()
1363 std::vector<ShapeHandle> input_tensors_as_shapes; in AddNode() local
1367 input_tensors, input_tensors_as_shapes, in AddNode()
1465 ic->input_tensors_as_shapes()[i]; in AllInputValuesKnown()
1541 const ShapeHandle& shape_handle = ic->input_tensors_as_shapes()[i]; in CreateInputTensors()
2679 } else if (static_cast<int>(ic->input_tensors_as_shapes().size()) > in InferStatically()
2682 ic, ic->input(i), ic->input_tensors_as_shapes()[i], in InferStatically()
2685 ic, ic->input(i), ic->input_tensors_as_shapes()[i], in InferStatically()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental.cc | 598 TF_ShapeAndTypeList* input_tensors_as_shapes, in TFE_InferShapes() argument
|