Searched defs:weight_store (Results 1 – 4 of 4) sorted by relevance
61 TrtWeightStore* weight_store; member
840 std::vector<TRT_TensorOrWeights>* outputs, TrtWeightStore* weight_store, in OpConverterParams()856 std::vector<TRT_TensorOrWeights>* outputs, TrtWeightStore* weight_store) in OpConverterParams()3434 Status TfTensorToTrtWeights(const Tensor& tensor, TrtWeightStore* weight_store, in TfTensorToTrtWeights()
731 Converter* converter, TrtWeightStore* weight_store, in TestPrepareTensorForShape()851 void TestGetWeightRange(ConverterTest* test, TrtWeightStore* weight_store) { in TestGetWeightRange()
44 nvinfer1::INetworkDefinition* network, TrtWeightStore* weight_store) { in Create()52 TrtWeightStore* weight_store) in TRTNetworkBuilder()