Home
last modified time | relevance | path

Searched refs:dims_input_tensor (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/
H A Dfill_ops.cc218 ITensorProxyPtr dims_input_tensor = nullptr; in Convert() local
236 dims_input_tensor = (*ceil_div)->getOutput(0); in Convert()
237 dims_input_tensor->setType(nvinfer1::DataType::kINT32); in Convert()
257 dims_input_tensor = (*const_layer)->getOutput(0); in Convert()
260 TRT_TensorOrWeights dims_input(dims_input_tensor); in Convert()
H A Dlike_ops.cc78 ITensorProxyPtr dims_input_tensor; in Convert() local
83 dims_input_tensor = (*shape_layer)->getOutput(0); in Convert()
87 TRT_TensorOrWeights dims_input(dims_input_tensor); in Convert()