Home
last modified time | relevance | path

Searched defs:input_tensors_as_shapes (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dshape_inference.h383 const std::vector<ShapeHandle>& 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
H A Dshape_inference.cc38 const std::vector<PartialTensorShape>& input_tensors_as_shapes, in InferenceContext()
90 const std::vector<ShapeHandle>& input_tensors_as_shapes, in InferenceContext()
168 const std::vector<ShapeHandle>& input_tensors_as_shapes) { in PreInputInit()
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/utils/
H A Dshape_inference_utils.cc289 std::vector<ShapeHandle> input_tensors_as_shapes(num_operands); in InferReturnTypeComponentsForTFOp() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dshape_refiner.cc677 std::vector<ShapeHandle> input_tensors_as_shapes; in RunShapeFn() local
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dc_api_experimental.cc606 TF_ShapeAndTypeList* input_tensors_as_shapes, in TFE_InferShapes()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dgraph_properties.cc1375 std::vector<ShapeHandle> input_tensors_as_shapes; in AddNode() local