Searched defs:weight_store (Results 1 – 3 of 3) sorted by relevance
376 TrtWeightStore* weight_store; member
1126 std::vector<TRT_TensorOrWeights>* outputs, TrtWeightStore* weight_store, in OpConverterParams()1141 std::vector<TRT_TensorOrWeights>* outputs, TrtWeightStore* weight_store) in OpConverterParams()3961 Status TfTensorToTrtWeights(const Tensor& tensor, TrtWeightStore* weight_store, in TfTensorToTrtWeights()
862 Converter* converter, TrtWeightStore* weight_store, in TestPrepareTensorForShape()975 void TestGetWeightRange(ConverterTest* test, TrtWeightStore* weight_store) { in TestGetWeightRange()