Searched defs:enable_moving_cpu_tensors_to_cuda (Results 1 – 2 of 2) sorted by relevance
80 generated_module_name, str_dict, enable_moving_cpu_tensors_to_cuda argument101 module_interface_cls, enable_moving_cpu_tensors_to_cuda=True argument
86 def _instantiate_template(module_interface_cls, enable_moving_cpu_tensors_to_cuda): argument483 def _init_template(self, module_interface_cls, enable_moving_cpu_tensors_to_cuda): argument