Searched defs:allclose (Results 1 – 7 of 7) sorted by relevance
/aosp_15_r20/external/pytorch/torch/distributed/_shard/sharded_tensor/_ops/ |
H A D | binary_cmp.py | 78 def allclose(types, args, kwargs, process_group): function
|
/aosp_15_r20/external/pytorch/test/lazy/ |
H A D | test_extract_compiled_graph.py | 113 def allclose(expected, actual): function
|
/aosp_15_r20/external/executorch/test/end2end/ |
H A D | test_end2end.py | 412 def allclose(lhs, rhs, rtol=1e-5, atol=1e-8): function
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
H A D | TensorCompare.cpp | 284 bool allclose(const Tensor& self, const Tensor& other, double rtol, double atol, bool equal_nan) { in allclose() function
|
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/numpy_ops/ |
H A D | np_math_ops.py | 481 def allclose(a, b, rtol=1e-05, atol=1e-08, equal_nan=False): function
|
/aosp_15_r20/external/pytorch/torch/_numpy/ |
H A D | _funcs_impl.py | 827 def allclose(a: ArrayLike, b: ArrayLike, rtol=1e-05, atol=1e-08, equal_nan=False): function
|
/aosp_15_r20/external/pytorch/torch/_refs/ |
H A D | __init__.py | 5647 def allclose( function
|