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