Searched refs:input_tensors_as_shapes (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | shape_inference_utils.cc | 275 std::vector<ShapeHandle> input_tensors_as_shapes; in InferReturnTypeComponentsForTFOp() local
283 input_tensors_as_shapes.resize(c.num_inputs()); in InferReturnTypeComponentsForTFOp()
287 if (handle.Handle()) input_tensors_as_shapes[input] = handle; in InferReturnTypeComponentsForTFOp()
294 if (!input_tensors_as_shapes.empty()) { in InferReturnTypeComponentsForTFOp()
295 c.set_input_tensors_as_shapes(input_tensors_as_shapes); in InferReturnTypeComponentsForTFOp()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.h | 173 const std::vector<ShapeHandle>& input_tensors_as_shapes,
189 const std::vector<PartialTensorShape>& input_tensors_as_shapes,
307 const std::vector<ShapeHandle>& input_tensors_as_shapes) { in set_input_tensors_as_shapes() argument
308 input_tensors_as_shapes_ = input_tensors_as_shapes; in set_input_tensors_as_shapes()
311 const std::vector<ShapeHandle>& input_tensors_as_shapes() const { in input_tensors_as_shapes() function
680 const std::vector<ShapeHandle>& input_tensors_as_shapes);
|
| D | shape_inference.cc | 37 const std::vector<PartialTensorShape>& input_tensors_as_shapes, in InferenceContext() argument
43 input_tensors_as_shape_handles.reserve(input_tensors_as_shapes.size()); in InferenceContext()
44 for (const PartialTensorShape& p : input_tensors_as_shapes) { in InferenceContext()
89 const std::vector<ShapeHandle>& input_tensors_as_shapes, in InferenceContext() argument
93 PreInputInit(op_def, input_tensors, input_tensors_as_shapes); in InferenceContext()
168 const std::vector<ShapeHandle>& input_tensors_as_shapes) { in PreInputInit() argument
170 input_tensors_as_shapes_ = input_tensors_as_shapes; in PreInputInit()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.cc | 656 std::vector<ShapeHandle> input_tensors_as_shapes; in RunShapeFn() local
661 c->set_input_tensors_as_shapes(input_tensors_as_shapes); in RunShapeFn()
745 if (i >= input_tensors_as_shapes.size()) { in RunShapeFn()
746 input_tensors_as_shapes.resize(i + 1); in RunShapeFn()
750 input_tensors_as_shapes[i] = s; in RunShapeFn()
759 c->set_input_tensors_as_shapes(input_tensors_as_shapes); in RunShapeFn()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 883 } else if (static_cast<int>(ic->input_tensors_as_shapes().size()) > i && in UpdateFunction()
885 ic, ic->input(i), ic->input_tensors_as_shapes()[i], in UpdateFunction()
890 ic, ic->input(i), ic->input_tensors_as_shapes()[i], in UpdateFunction()
1337 std::vector<ShapeHandle> input_tensors_as_shapes; in AddNode() local
1341 input_tensors, input_tensors_as_shapes, in AddNode()
1438 ic->input_tensors_as_shapes()[i]; in AllInputValuesKnown()
1514 const ShapeHandle& shape_handle = ic->input_tensors_as_shapes()[i]; in CreateInputTensors()
2647 } else if (static_cast<int>(ic->input_tensors_as_shapes().size()) > in InferStatically()
2650 ic, ic->input(i), ic->input_tensors_as_shapes()[i], in InferStatically()
2653 ic, ic->input(i), ic->input_tensors_as_shapes()[i], in InferStatically()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental.cc | 599 TF_ShapeAndTypeList* input_tensors_as_shapes, in TFE_InferShapes() argument
|