Searched defs:to_device (Results 1 – 5 of 5) sorted by relevance
123 bool to_device = !(flags & CL_MIGRATE_MEM_OBJECT_HOST); in svm_migrate() local
39 bool to_device, bool mem_undefined) in nvc0_svm_migrate()
145 … .def("__deepcopy__", [](py::object &to_device, py::dict memo) { return to_device; }); in __anona1cf9bfd1702()
1969 bool NchwFracZTransWithGroups(const FormatArgs &args, void *result, bool to_device, int64_t groups)… in NchwFracZTransWithGroups()
1291 def to_device(self, send_epoch_end=True, create_data_info_queue=False): member in Dataset