Searched refs:trt_dims (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
D | utils.h | 143 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));
|
D | utils.cc | 195 Status TrtDimsToTensorShape(const std::vector<int>& trt_dims, in TrtDimsToTensorShape() argument 199 TensorShapeUtils::MakeShape(trt_dims.data(), trt_dims.size(), shape)); in TrtDimsToTensorShape()
|
D | convert_nodes.cc | 309 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 …]
|
D | convert_nodes.h | 278 const nvinfer1::Dims& trt_dims, int batch_size);
|
D | convert_nodes_test.cc | 1544 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/ |
D | trt_engine_utils.cc | 164 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()
|