MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / GetTrtWeights

Method GetTrtWeights

tensorflow/compiler/tf2tensorrt/convert/convert_nodes.cc:680–682  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

678}
679
680nvinfer1::Weights TRT_ShapedWeights::GetTrtWeights() const {
681 return nvinfer1::Weights{type_, GetValues(), count()};
682}
683
684size_t TRT_ShapedWeights::size_bytes() const {
685 size_t data_type_size = -1;

Callers 8

TESTFunction · 0.80
CreateConstantLayerMethod · 0.80
ConvertConv2DHelperFunction · 0.80
ConvertConv3DHelperFunction · 0.80
ConvertFusedBatchNormFunction · 0.80

Calls 2

GetValuesFunction · 0.85
countFunction · 0.85

Tested by 1

TESTFunction · 0.64