Searched refs:GetTrtWeights (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 472 nvinfer1::Weights trt_weights = weights.GetTrtWeights(); in CreateConstantLayer()
705 nvinfer1::Weights TRT_ShapedWeights::GetTrtWeights() const { in GetTrtWeights() function in tensorflow::tensorrt::convert::TRT_ShapedWeights
2396 *tensor, noutput, kernel_size, weights.GetTrtWeights(), in ConvertConv2DHelper()
2397 biases.GetTrtWeights()); in ConvertConv2DHelper()
2415 *tensor, noutput, kernel_size, weights.GetTrtWeights(), in ConvertConv2DHelper()
2416 biases.GetTrtWeights()); in ConvertConv2DHelper()
3365 *tensor, noutput, kernel_size_drs, weights.GetTrtWeights(), in ConvertConv3DHelper()
3366 biases.GetTrtWeights()); in ConvertConv3DHelper()
3382 *tensor, noutput, kernel_size_drs, weights.GetTrtWeights(), in ConvertConv3DHelper()
3383 biases.GetTrtWeights()); in ConvertConv3DHelper()
[all …]
|
| D | convert_nodes.h | 181 nvinfer1::Weights GetTrtWeights() const;
|
| D | convert_nodes_test.cc | 406 nvinfer1::Weights trt_weights = ptr->GetTrtWeights(); in TEST()
421 nvinfer1::Weights trt_weights = ptr->GetTrtWeights(); in TEST()
438 nvinfer1::Weights trt_weights = ptr->GetTrtWeights(); in TEST()
|