Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dutils.h143 nvinfer1::Dims trt_dims; in TensorShapeToTrtDims() local
146 trt_dims.d[i - offset] = shape.dim_size(i); in TensorShapeToTrtDims()
148 trt_dims.nbDims = shape.dims() - offset; in TensorShapeToTrtDims()
149 return trt_dims; in TensorShapeToTrtDims()
155 Status TrtDimsToTensorShape(const std::vector<int>& trt_dims,
160 Status TrtDimsToTensorShape(const nvinfer1::Dims trt_dims,
164 TensorShapeUtils::MakeShape(trt_dims.d, trt_dims.nbDims, shape));
Dutils.cc195 Status TrtDimsToTensorShape(const std::vector<int>& trt_dims, in TrtDimsToTensorShape() argument
199 TensorShapeUtils::MakeShape(trt_dims.data(), trt_dims.size(), shape)); in TrtDimsToTensorShape()
Dconvert_nodes.cc309 nvinfer1::Dims* trt_dims, int* batch_size) { in ValidateTensorProperties() argument
328 *trt_dims = TensorShapeToTrtDims(shape, in ValidateTensorProperties()
766 SimpleITensor(nvinfer1::DataType trt_dtype, const nvinfer1::Dims& trt_dims) in SimpleITensor() argument
767 : trt_dtype_(trt_dtype), trt_dims_(trt_dims) {} in SimpleITensor()
839 const nvinfer1::Dims& trt_dims, in TRT_TensorOrWeights() argument
841 : simple_itensor_(new SimpleITensor(trt_dtype, trt_dims)), in TRT_TensorOrWeights()
1175 nvinfer1::Dims trt_dims; in ConvertToTensorOrWeights() local
1179 /*validation_only_=*/true, &trt_dtype, &trt_dims, &batch_size)); in ConvertToTensorOrWeights()
1184 *tensor_or_weights = TRT_TensorOrWeights(trt_dtype, trt_dims, batch_size); in ConvertToTensorOrWeights()
5698 const nvinfer1::Dims trt_dims = params->inputs.at(0).GetTrtDims(); in ConvertArgMinMax() local
[all …]
Dconvert_nodes.h278 const nvinfer1::Dims& trt_dims, int batch_size);
Dconvert_nodes_test.cc1544 const nvinfer1::Dims trt_dims = in AddTestTensorWithTFDims() local
1546 if (!converter_->use_implicit_batch() || HasStaticShape(trt_dims)) { in AddTestTensorWithTFDims()
1549 converter_->AddInputTensor(name, trt_type, trt_dims, batch_size); in AddTestTensorWithTFDims()
1585 const nvinfer1::Dims trt_dims = GetTestDims(dims); in AddTestWeights() local
1586 const int64_t num_elements = TrtWeightDimsNumElements(trt_dims); in AddTestWeights()
1591 weights = converter_->weight_store_.GetTempWeights(dtype, trt_dims); in AddTestWeights()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_engine_utils.cc164 nvinfer1::Dims trt_dims; in SetTrtEngineInputs() local
165 trt_dims.nbDims = input_shape.dims(); in SetTrtEngineInputs()
167 trt_dims.d[k] = input_shape.dim_size(k); in SetTrtEngineInputs()
170 execution_context->setBindingDimensions(binding_index, trt_dims); in SetTrtEngineInputs()
173 << DebugString(trt_dims); in SetTrtEngineInputs()