Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
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
680
nvinfer1::Weights TRT_ShapedWeights::GetTrtWeights() const {
681
return nvinfer1::Weights{type_, GetValues(), count()};
682
}
683
684
size_t TRT_ShapedWeights::size_bytes() const {
685
size_t data_type_size = -1;
Callers
8
TEST
Function · 0.80
CreateConstantLayer
Method · 0.80
ConvertConv2DHelper
Function · 0.80
ConvertConv3DHelper
Function · 0.80
ConvertFusedConv2DBiasActivation
Function · 0.80
ConvertBiasAddInt8WithoutCalibration
Function · 0.80
ConvertFusedBatchNorm
Function · 0.80
ConvertFullyConnectedHelper
Function · 0.80
Calls
2
GetValues
Function · 0.85
count
Function · 0.85
Tested by
1
TEST
Function · 0.64