Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc643 class TRT_TensorOrWeights::SimpleITensor : public nvinfer1::ITensor { class in tensorflow::tensorrt::convert::TRT_TensorOrWeights
645 SimpleITensor(nvinfer1::DataType trt_dtype, const nvinfer1::Dims& trt_dims) in SimpleITensor() function in tensorflow::tensorrt::convert::TRT_TensorOrWeights::SimpleITensor