Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/framework/
Dshape_inference.h307 const std::vector<ShapeHandle>& 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
Dshape_inference.cc37 const std::vector<PartialTensorShape>& input_tensors_as_shapes, in InferenceContext()
89 const std::vector<ShapeHandle>& input_tensors_as_shapes, in InferenceContext()
168 const std::vector<ShapeHandle>& input_tensors_as_shapes) { in PreInputInit()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dshape_inference_utils.cc275 std::vector<ShapeHandle> input_tensors_as_shapes; in InferReturnTypeComponentsForTFOp() local
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner.cc656 std::vector<ShapeHandle> input_tensors_as_shapes; in RunShapeFn() local
/external/tensorflow/tensorflow/c/
Dc_api_experimental.cc599 TF_ShapeAndTypeList* input_tensors_as_shapes, in TFE_InferShapes()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc1337 std::vector<ShapeHandle> input_tensors_as_shapes; in AddNode() local