Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.h311 class SimpleITensor;
329 std::shared_ptr<SimpleITensor> simple_itensor_ = nullptr;
Dconvert_nodes.cc764 class TRT_TensorOrWeights::SimpleITensor : public nvinfer1::ITensor { class in tensorflow::tensorrt::convert::TRT_TensorOrWeights
766 SimpleITensor(nvinfer1::DataType trt_dtype, const nvinfer1::Dims& trt_dims) in SimpleITensor() function in tensorflow::tensorrt::convert::TRT_TensorOrWeights::SimpleITensor
841 : simple_itensor_(new SimpleITensor(trt_dtype, trt_dims)), in TRT_TensorOrWeights()