Searched refs:sparseTensorSetToDeviceType (Results 1 – 1 of 1) sorted by relevance
9 DeviceType sparseTensorSetToDeviceType(DispatchKeySet key_set) { in sparseTensorSetToDeviceType() function32 …, at::empty({1, 0}, at::initialTensorOptions().device(sparseTensorSetToDeviceType(key_set)).dtype(… in SparseTensorImpl()33 …, at::empty({0}, at::initialTensorOptions().device(sparseTensorSetToDeviceType(key_set)).dtype(dat…