Searched defs:weight_store (Results 1 – 3 of 3) sorted by relevance
378 TrtWeightStore* weight_store; member
1103 std::vector<TRT_TensorOrWeights>* outputs, TrtWeightStore* weight_store, in OpConverterParams()1118 std::vector<TRT_TensorOrWeights>* outputs, TrtWeightStore* weight_store) in OpConverterParams()4307 Status TfTensorToTrtWeights(const Tensor& tensor, TrtWeightStore* weight_store, in TfTensorToTrtWeights()
939 Converter* converter, TrtWeightStore* weight_store, in TestPrepareTensorForShape()1052 void TestGetWeightRange(ConverterTest* test, TrtWeightStore* weight_store) { in TestGetWeightRange()