Home
last modified time | relevance | path

Searched refs:requested_input_tensor (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/ir/utils/
H A Dshape_inference_utils.cc296 (c.requested_input_tensor(input) || in InferReturnTypeComponentsForTFOp()
311 if (c.requested_input_tensor(input)) { in InferReturnTypeComponentsForTFOp()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dshape_inference.h358 bool requested_input_tensor(int idx) const { in requested_input_tensor() function
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dshape_refiner.cc743 if (!c->requested_input_tensor(i)) { in RunShapeFn()