Searched defs:float_module_class (Results 1 – 2 of 2) sorted by relevance
79 float_module_class, argument274 self, float_module_class, quantized_module_class, extra_module_kwargs): argument
1152 …def _test_activation_module_impl(self, name, float_module_class, quantized_module_class, extra_kwa… argument