Searched refs:TrtWeightStore (Results 1 – 3 of 3) sorted by relevance
224 friend class TrtWeightStore; variable234 class TrtWeightStore {351 bool arg_validation_only, TrtWeightStore* arg_weight_store) in OpConverterParams()364 TrtWeightStore* weight_store;413 TrtWeightStore weight_store_;553 TrtWeightStore weight_store_;
301 TrtWeightStore store; in TEST()576 TrtWeightStore* weight_store_;803 void TestGetWeightRange(ConverterTest* test, TrtWeightStore* weight_store) { in TestGetWeightRange()
872 TRT_ShapedWeights TrtWeightStore::GetTempWeights(DataType type, in GetTempWeights()1480 TRT_ShapedWeights ConvertFP32ToFP16(TrtWeightStore* store, in ConvertFP32ToFP16()3041 Status TfTensorToTrtWeights(const Tensor& tensor, TrtWeightStore* weight_store, in TfTensorToTrtWeights()