MCPcopy Create free account
hub / github.com/PythonOT/POT / dtype_device

Method dtype_device

ot/backend.py:3216–3217  ·  view source on GitHub ↗
(self, a)

Source from the content-addressed store, hash-verified

3214 return tnp.allclose(a, b, rtol=rtol, atol=atol, equal_nan=equal_nan)
3215
3216 def dtype_device(self, a):
3217 return a.dtype, a.device.split("device:")[1]
3218
3219 def assert_same_dtype_device(self, a, b):
3220 a_dtype, a_device = self.dtype_device(a)

Callers 10

assert_same_dtype_deviceMethod · 0.95
device_typeMethod · 0.95
test_sinkhorn2_variants_device_tfFunction · 0.95
test_wasserstein_bary_2d_device_tfFunction · 0.95
test_sliced_backend_device_tfFunction · 0.95
test_max_sliced_backend_device_tfFunction · 0.95
test_wasserstein_1d_device_tfFunction · 0.95
test_emd1d_device_tfFunction · 0.95
test_emd_emd2_devices_tfFunction · 0.95
test_gromov_device_tfFunction · 0.95

Calls

no outgoing calls

Tested by 8

test_sinkhorn2_variants_device_tfFunction · 0.76
test_wasserstein_bary_2d_device_tfFunction · 0.76
test_sliced_backend_device_tfFunction · 0.76
test_max_sliced_backend_device_tfFunction · 0.76
test_wasserstein_1d_device_tfFunction · 0.76
test_emd1d_device_tfFunction · 0.76
test_emd_emd2_devices_tfFunction · 0.76
test_gromov_device_tfFunction · 0.76