Searched refs:fp32_module (Results 1 – 1 of 1) sorted by relevance
214 def _test_package(self, fp32_module, input_size, generate=False): argument258 exp.save_pickle(package_name, resource_name_model, fp32_module)261 mq = _do_quant_transforms(fp32_module, input_tensor)