Searched refs:trt_dims (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
D | convert_nodes.cc | 204 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 …]
|
D | convert_nodes.h | 270 const nvinfer1::Dims& trt_dims, int batch_size);
|
D | convert_nodes_test.cc | 1205 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()
|