(self, a, b, rtol=1e-05, atol=1e-08, equal_nan=False)
| 3211 | return tf.identity(a) |
| 3212 | |
| 3213 | def allclose(self, a, b, rtol=1e-05, atol=1e-08, equal_nan=False): |
| 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] |