Home
last modified time | relevance | path

Searched defs:allclose (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/_shard/sharded_tensor/_ops/
H A Dbinary_cmp.py78 def allclose(types, args, kwargs, process_group): function
/aosp_15_r20/external/pytorch/test/lazy/
H A Dtest_extract_compiled_graph.py113 def allclose(expected, actual): function
/aosp_15_r20/external/executorch/test/end2end/
H A Dtest_end2end.py412 def allclose(lhs, rhs, rtol=1e-5, atol=1e-8): function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorCompare.cpp284 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 Dnp_math_ops.py481 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.py827 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__.py5647 def allclose( function