Searched defs:module_interface_cls (Results 1 – 2 of 2) sorted by relevance
86 def _instantiate_template(module_interface_cls, enable_moving_cpu_tensors_to_cuda): argument104 module_cls, args, kwargs, device, module_interface_cls argument483 def _init_template(self, module_interface_cls, enable_moving_cpu_tensors_to_cuda): argument
101 module_interface_cls, enable_moving_cpu_tensors_to_cuda=True argument