Searched refs:GetTrtWeights (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 447 nvinfer1::Weights trt_weights = weights.GetTrtWeights(); in CreateConstantLayer()
624 nvinfer1::Weights TRT_ShapedWeights::GetTrtWeights() const { in GetTrtWeights() function in tensorflow::tensorrt::convert::TRT_ShapedWeights
1745 shift_weights.GetTrtWeights(), scale_weights.GetTrtWeights(), in BinaryTensorOpWeight()
1746 power_weights.GetTrtWeights()); in BinaryTensorOpWeight()
1902 weights.GetTrtWeights(), biases.GetTrtWeights()); in ConvertConv2DHelper()
1913 weights.GetTrtWeights(), biases.GetTrtWeights()); in ConvertConv2DHelper()
2993 *tensor, mode, weights.GetTrtWeights(), empty_weights.GetTrtWeights(), in ConvertBiasAdd()
2994 empty_weights.GetTrtWeights()); in ConvertBiasAdd()
3703 combined_offset_weights.GetTrtWeights(), in ConvertFusedBatchNorm()
3704 combined_scale_weights.GetTrtWeights(), in ConvertFusedBatchNorm()
[all …]
|
| D | convert_nodes.h | 184 nvinfer1::Weights GetTrtWeights() const;
|
| D | convert_nodes_test.cc | 274 nvinfer1::Weights trt_weights = ptr->GetTrtWeights(); in TEST()
289 nvinfer1::Weights trt_weights = ptr->GetTrtWeights(); in TEST()
306 nvinfer1::Weights trt_weights = ptr->GetTrtWeights(); in TEST()
|