Home
last modified time | relevance | path

Searched refs:requested_input_tensor_as_partial_shape (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dshape_inference_utils.cc262 c.requested_input_tensor_as_partial_shape(input)); in InferReturnTypeComponentsForTFOp()
295 if (c.requested_input_tensor_as_partial_shape(input) && in InferReturnTypeComponentsForTFOp()
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner.cc353 if (node_context->requested_input_tensor_as_partial_shape(dst_input)) { in UpdateNode()
759 if (c->requested_input_tensor_as_partial_shape(i) && in RunShapeFn()
/external/tensorflow/tensorflow/core/framework/
Dshape_inference.h300 bool requested_input_tensor_as_partial_shape(int idx) const { in requested_input_tensor_as_partial_shape() function
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc1041 if (!*refined && ic->requested_input_tensor_as_partial_shape(dst_input)) { in UpdateNode()