Home
last modified time | relevance | path

Searched refs:trt_dims (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc204 nvinfer1::Dims trt_dims; in TensorShapeToTrtDims() local
207 trt_dims.d[i - offset] = shape.dim_size(i); in TensorShapeToTrtDims()
209 trt_dims.nbDims = shape.dims() - offset; in TensorShapeToTrtDims()
210 return trt_dims; in TensorShapeToTrtDims()
265 nvinfer1::Dims* trt_dims, int* batch_size) { in ValidateTensorProperties() argument
282 *trt_dims = TensorShapeToTrtDims(shape, /*ignore_first_dim=*/true); in ValidateTensorProperties()
645 SimpleITensor(nvinfer1::DataType trt_dtype, const nvinfer1::Dims& trt_dims) in SimpleITensor() argument
646 : trt_dtype_(trt_dtype), trt_dims_(trt_dims) {} in SimpleITensor()
708 const nvinfer1::Dims& trt_dims, in TRT_TensorOrWeights() argument
710 : simple_itensor_(new SimpleITensor(trt_dtype, trt_dims)), in TRT_TensorOrWeights()
[all …]
Dconvert_nodes.h270 const nvinfer1::Dims& trt_dims, int batch_size);
Dconvert_nodes_test.cc1205 const nvinfer1::Dims trt_dims = GetTestDims(dims); in AddTestTensor() local
1206 if (HasStaticShape(trt_dims)) { in AddTestTensor()
1208 converter_->AddInputTensor(name, trt_dtype, trt_dims, batch_size)); in AddTestTensor()
1218 const nvinfer1::Dims trt_dims = GetTestDims(dims); in AddTestWeights() local
1219 const int64_t num_elements = TrtDimsNumElements(trt_dims); in AddTestWeights()
1224 weights = converter_->weight_store_.GetTempWeights(dtype, trt_dims); in AddTestWeights()