Home
last modified time | relevance | path

Searched defs:trt_tensor (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
H A Dtrt_tensor_proxy.h118 ITensorProxy(nvinfer1::ITensor* trt_tensor) in ITensorProxy()
157 nvinfer1::ITensor* trt_tensor() const { in trt_tensor() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes.cc485 ITensorProxyPtr trt_tensor = layer->getOutput(0); in CreateConstantLayer() local