Searched refs:input_tensors_as_shapes (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/ir/utils/ |
| D | shape_inference_utils.cc | 289 std::vector<ShapeHandle> input_tensors_as_shapes(num_operands); in InferReturnTypeComponentsForTFOp() local
329 !input_tensors[input] && !input_tensors_as_shapes[input].Handle()) { in InferReturnTypeComponentsForTFOp()
338 input_tensors_as_shapes[input] = handle; in InferReturnTypeComponentsForTFOp()
347 c.set_input_tensors_as_shapes(input_tensors_as_shapes); in InferReturnTypeComponentsForTFOp()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.h | 249 const std::vector<ShapeHandle>& input_tensors_as_shapes,
265 const std::vector<PartialTensorShape>& input_tensors_as_shapes,
383 const std::vector<ShapeHandle>& input_tensors_as_shapes) { in set_input_tensors_as_shapes() argument
384 input_tensors_as_shapes_ = input_tensors_as_shapes; in set_input_tensors_as_shapes()
387 const std::vector<ShapeHandle>& input_tensors_as_shapes() const { in input_tensors_as_shapes() function
766 const std::vector<ShapeHandle>& input_tensors_as_shapes);
|
| D | shape_inference.cc | 38 const std::vector<PartialTensorShape>& input_tensors_as_shapes, in InferenceContext() argument
44 input_tensors_as_shape_handles.reserve(input_tensors_as_shapes.size()); in InferenceContext()
45 for (const PartialTensorShape& p : input_tensors_as_shapes) { in InferenceContext()
90 const std::vector<ShapeHandle>& input_tensors_as_shapes, in InferenceContext() argument
94 PreInputInit(op_def, input_tensors, input_tensors_as_shapes); in InferenceContext()
168 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/core/common_runtime/ |
| D | shape_refiner.cc | 677 std::vector<ShapeHandle> input_tensors_as_shapes; in RunShapeFn() local
682 c->set_input_tensors_as_shapes(input_tensors_as_shapes); in RunShapeFn()
766 if (i >= input_tensors_as_shapes.size()) { in RunShapeFn()
767 input_tensors_as_shapes.resize(i + 1); in RunShapeFn()
771 input_tensors_as_shapes[i] = s; in RunShapeFn()
780 c->set_input_tensors_as_shapes(input_tensors_as_shapes); in RunShapeFn()
|
| /external/tensorflow/tensorflow/security/advisory/ |
| D | tfsa-2022-035.md | 12 const std::vector<ShapeHandle>& input_tensors_as_shapes) {
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 912 } else if (static_cast<int>(ic->input_tensors_as_shapes().size()) > i && in UpdateFunction()
914 ic, ic->input(i), ic->input_tensors_as_shapes()[i], in UpdateFunction()
919 ic, ic->input(i), ic->input_tensors_as_shapes()[i], in UpdateFunction()
1375 std::vector<ShapeHandle> input_tensors_as_shapes; in AddNode() local
1379 input_tensors, input_tensors_as_shapes, in AddNode()
1477 ic->input_tensors_as_shapes()[i]; in AllInputValuesKnown()
1553 const ShapeHandle& shape_handle = ic->input_tensors_as_shapes()[i]; in CreateInputTensors()
2691 } else if (static_cast<int>(ic->input_tensors_as_shapes().size()) > in InferStatically()
2694 ic, ic->input(i), ic->input_tensors_as_shapes()[i], in InferStatically()
2697 ic, ic->input(i), ic->input_tensors_as_shapes()[i], in InferStatically()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental.cc | 606 TF_ShapeAndTypeList* input_tensors_as_shapes, in TFE_InferShapes() argument
|