Searched refs:set_python_custom_device (Results 1 – 2 of 2) sorted by relevance
726 data.unsafeGetTensorImpl()->set_python_custom_device(true); in THPVariable_make_subclass()877 tensor.unsafeGetTensorImpl()->set_python_custom_device(true); in THPVariable_make_wrapper_subclass()
2803 void set_python_custom_device(bool custom_device) { in set_python_custom_device() function