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.h309 const std::vector<ShapeHandle>& 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
Dshape_inference.cc39 const std::vector<PartialTensorShape>& input_tensors_as_shapes, in InferenceContext()
91 const std::vector<ShapeHandle>& input_tensors_as_shapes, in InferenceContext()
170 const std::vector<ShapeHandle>& input_tensors_as_shapes) { in PreInputInit()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dshape_inference_utils.cc254 std::vector<ShapeHandle> input_tensors_as_shapes(num_operands); in InferReturnTypeComponentsForTFOp() local
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner.cc673 std::vector<ShapeHandle> input_tensors_as_shapes; in RunShapeFn() local
/external/tensorflow/tensorflow/c/
Dc_api_experimental.cc598 TF_ShapeAndTypeList* input_tensors_as_shapes, in TFE_InferShapes()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc1363 std::vector<ShapeHandle> input_tensors_as_shapes; in AddNode() local