Searched refs:trt_dims_ (Results 1 – 1 of 1) sorted by relevance
646 : trt_dtype_(trt_dtype), trt_dims_(trt_dims) {} in SimpleITensor()653 trt_dims_ = dimensions; in setDimensions()656 nvinfer1::Dims getDimensions() const override { return trt_dims_; } in getDimensions()697 nvinfer1::Dims trt_dims_; member in tensorflow::tensorrt::convert::TRT_TensorOrWeights::SimpleITensor