Searched refs:fake_quant_table (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | pipeline.cc | 523 std::map<std::string, std::pair<PrimitivePyAdapterPtr, std::string>> *fake_quant_table) { in GetWeightInfo() argument
525 MS_EXCEPTION_IF_NULL(fake_quant_table); in GetWeightInfo()
558 (*fake_quant_table)[weight_name] = std::make_pair(nullptr, "input"); in GetWeightInfo()
588 (*fake_quant_table)[weight_name] = std::make_pair(quant_op->adapter(), fakequant_min_node_name); in GetWeightInfo()
596 std::map<std::string, std::pair<PrimitivePyAdapterPtr, std::string>> fake_quant_table; in FetchInfoForQuantExport() local
634 GetWeightInfo(root_node, weight_node, &fake_quant_table); in FetchInfoForQuantExport()
636 return fake_quant_table; in FetchInfoForQuantExport()
|
| D | pipeline.h | 126 … std::map<std::string, std::pair<PrimitivePyAdapterPtr, std::string>> *fake_quant_table);
|