Home
last modified time | relevance | path

Searched full:allclose (Results 1 – 25 of 334) sorted by relevance

12345678910>>...14

/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dfunctional.cpp26 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
29 ASSERT_TRUE(torch::allclose(y_no_options, expected)); in TEST_F()
48 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
51 ASSERT_TRUE(torch::allclose(y_no_options, expected)); in TEST_F()
68 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
71 ASSERT_TRUE(torch::allclose(y_no_options, expected)); in TEST_F()
106 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
109 ASSERT_TRUE(torch::allclose(y_no_options, expected)); in TEST_F()
117 ASSERT_TRUE(torch::allclose(y, torch::ones({1, 1, 2}))); in TEST_F()
126 ASSERT_TRUE(torch::allclose(y, torch::ones({2, 2, 2}))); in TEST_F()
[all …]
H A Dmodules.cpp53 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
92 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
114 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
168 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
202 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
237 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
273 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
293 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
309 ASSERT_TRUE(torch::allclose(y, torch::ones({1, 1, 2}))); in TEST_F()
320 ASSERT_TRUE(torch::allclose(y, torch::ones({1, 1, 2}))); in TEST_F()
[all …]
H A Dnn_utils.cpp107 ASSERT_TRUE(torch::allclose(p1.grad(), p2.grad())); in TEST_F()
257 ASSERT_TRUE(torch::allclose( in TEST_F()
345 ASSERT_TRUE(torch::allclose(p1.grad(), p2.grad())); in TEST_F()
359 ASSERT_TRUE(vector.allclose(expected)); in TEST_F()
368 ASSERT_TRUE(zero_parameters[i].allclose(parameters[i])); in TEST_F()
434 ASSERT_TRUE(torch::allclose(a.data(), b.data())); in assert_is_equal_packed_sequence()
435 ASSERT_TRUE(torch::allclose(a.batch_sizes(), b.batch_sizes())); in assert_is_equal_packed_sequence()
438 torch::allclose(a.sorted_indices(), b.sorted_indices())); in assert_is_equal_packed_sequence()
441 torch::allclose(a.unsorted_indices(), b.unsorted_indices())); in assert_is_equal_packed_sequence()
500 ASSERT_TRUE(torch::allclose(lengths, lengths_out)); in TEST_F()
[all …]
H A Dtransformer.cpp74 torch::allclose(result, ref_output, 1e-7, 1e-5, /*equal_nan=*/true)); in transformer_encoder_layer_test_helper()
85 torch::allclose(result, ref_output, 1e-7, 1e-5, /*equal_nan=*/true)); in transformer_encoder_layer_test_helper()
107 torch::allclose(result, ref_output, 1e-7, 1e-5, /*equal_nan=*/true)); in transformer_encoder_layer_test_helper()
118 torch::allclose(result, ref_output, 1e-7, 1e-5, /*equal_nan=*/true)); in transformer_encoder_layer_test_helper()
133 torch::allclose(result, ref_output, 1e-7, 1e-5, /*equal_nan=*/true)); in transformer_encoder_layer_test_helper()
158 torch::allclose(result, ref_output, 1e-7, 1e-5, /*equal_nan=*/true)); in transformer_encoder_layer_test_helper()
169 torch::allclose(result, ref_output, 1e-7, 1e-5, /*equal_nan=*/true)); in transformer_encoder_layer_test_helper()
194 torch::allclose(result, ref_output, 1e-7, 1e-5, /*equal_nan=*/true)); in transformer_encoder_layer_test_helper()
204 torch::allclose(result, ref_output, 1e-7, 1e-5, /*equal_nan=*/true)); in transformer_encoder_layer_test_helper()
229 torch::allclose(result, ref_output, 1e-7, 1e-5, /*equal_nan=*/true)); in transformer_encoder_layer_test_helper()
[all …]
H A Dfft.cpp40 ASSERT_TRUE(torch::allclose(actual, expect)); in TEST()
47 ASSERT_TRUE(torch::allclose(actual, expect)); in TEST()
54 ASSERT_TRUE(torch::allclose(actual, expect)); in TEST()
58 ASSERT_TRUE(torch::allclose(actual, expect)); in TEST()
67 ASSERT_TRUE(torch::allclose(unnorm / 128, norm)); in TEST()
71 ASSERT_TRUE(torch::allclose(unnorm / std::sqrt(128), ortho_norm)); in TEST()
78 ASSERT_TRUE(torch::allclose(actual, expect)); in TEST()
90 ASSERT_TRUE(torch::allclose(t, t_round_trip)); in TEST()
97 ASSERT_TRUE(torch::allclose(actual, expect)); in TEST()
109 ASSERT_TRUE(torch::allclose(t, t_round_trip)); in TEST()
[all …]
H A Dsequential.cpp363 ASSERT_TRUE(param->allclose(params2[param.key()])); in TEST_F()
367 ASSERT_FALSE(param->allclose(params2[param.key()])); in TEST_F()
445 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
470 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
487 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
495 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
533 torch::allclose(attn_output, attn_output_expected, 1e-05, 2e-04)); in TEST_F()
545 ASSERT_TRUE(torch::allclose( in TEST_F()
555 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
581 ASSERT_TRUE(torch::allclose(y, expected)); in TEST_F()
[all …]
/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_dims.py142 self.assertTrue(torch.allclose(C.order(i, j), torch.mm(A_, B_)))
143 self.assertTrue(torch.allclose(torch.triu(A_, 0), triu(A_)))
184 torch.allclose(a_out, b_out)
218 self.assertTrue(torch.allclose(a_out, b_out))
267 self.assertTrue(torch.allclose(a_out, b_out))
315 # self.assertTrue(torch.allclose(a.order(i, j), i.expand(j).order(i, j)))
316 # self.assertTrue(torch.allclose(b.order(i, j), j.expand(i).order(i, j)))
323 self.assertTrue(torch.allclose(r.order(i, j), ap.max(2)[0]))
343 self.assertTrue(torch.allclose(a, x))
352 assert torch.allclose(r, A @ B)
[all …]
/aosp_15_r20/external/pytorch/test/cpp/lazy/
H A Dtest_lazy_ops.cpp92 AllClose(scalar_tensor, lazy_scalar_tensor); in TEST_F()
102 AllClose(a, lazy_b); in TEST_F()
104 AllClose(a, lazy_b); in TEST_F()
113 AllClose(a, lazy_a); in TEST_F()
251 AllClose(c, lazy_c); in TEST_F()
265 AllClose(c, lazy_c); in TEST_F()
279 AllClose(c, lazy_c); in TEST_F()
293 AllClose(a, lazy_a); in TEST_F()
294 AllClose(c, lazy_c); in TEST_F()
306 AllClose(c, lazy_c); in TEST_F()
[all …]
/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_await.py45 self.assertTrue(torch.allclose(torch.eye(2) + 102, script_out))
46 self.assertTrue(torch.allclose(script_out, out))
60 self.assertTrue(torch.allclose(torch.eye(2) + 13, script_out))
61 self.assertTrue(torch.allclose(script_out, out))
84 self.assertTrue(torch.allclose(torch.eye(2), script_out))
85 self.assertTrue(torch.allclose(script_out, out))
113 self.assertTrue(torch.allclose(torch.eye(2), script_out))
114 self.assertTrue(torch.allclose(script_out, out))
142 self.assertTrue(torch.allclose(torch.eye(2) + 2 * torch.ones(2), script_out))
143 self.assertTrue(torch.allclose(script_out, out))
[all …]
/aosp_15_r20/external/pytorch/test/cpp/tensorexpr/
H A Dtest_external_calls.cpp77 ASSERT_TRUE(at::allclose(nnc_result, ref)); in TEST()
84 ASSERT_TRUE(at::allclose(nnc_result, ref)); in TEST()
131 ASSERT_TRUE(at::allclose(nnc_result, ref)); in TEST()
138 ASSERT_TRUE(at::allclose(nnc_result, ref)); in TEST()
172 ASSERT_TRUE(at::allclose(nnc_result, ref)); in TEST()
179 ASSERT_TRUE(at::allclose(nnc_result, ref)); in TEST()
231 ASSERT_TRUE(at::allclose(nnc_result, ref)); in TEST()
238 ASSERT_TRUE(at::allclose(nnc_result, ref)); in TEST()
292 ASSERT_TRUE(at::allclose(nnc_result, ref)); in TEST()
299 ASSERT_TRUE(at::allclose(nnc_result, ref)); in TEST()
[all …]
/aosp_15_r20/external/pytorch/test/cpp/aoti_inference/
H A Dtest.cpp50 ASSERT_TRUE(torch::allclose(ref_output_tensors[0], actual_output_tensors[0])); in test_aoti()
76 ASSERT_TRUE(torch::allclose(outputs[i].toTensor(), ref_output_tensors[i])); in test_aoti_script()
129 ASSERT_TRUE(torch::allclose(ref_output_tensors[0], actual_output_tensors[0])); in test_aoti_constants_update()
143 torch::allclose(ref_output_tensors[0], actual_output_tensors[0])); in test_aoti_constants_update()
148 torch::allclose(ref_output_tensors[0], actual_output_tensors[0])); in test_aoti_constants_update()
157 ASSERT_TRUE(torch::allclose(ref_output_tensors[0], actual_output_tensors[0])); in test_aoti_constants_update()
166 torch::allclose(ref_output_tensors[0], actual_output_tensors[0])); in test_aoti_constants_update()
217 ASSERT_TRUE(torch::allclose(ref_output_tensors[0], actual_output_tensors[0])); in test_aoti_double_buffering()
227 ASSERT_TRUE(torch::allclose(ref_output_tensors[0], actual_output_tensors[0])); in test_aoti_double_buffering()
235 ASSERT_TRUE(torch::allclose(ref_output_tensors[0], actual_output_tensors[0])); in test_aoti_double_buffering()
[all …]
/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dcpu_rng_test.cpp214 ASSERT_TRUE(torch::allclose(actual, expected)); in TEST_F()
228 ASSERT_TRUE(torch::allclose(actual, expected)); in TEST_F()
242 ASSERT_TRUE(torch::allclose(actual, expected)); in TEST_F()
256 ASSERT_TRUE(torch::allclose(actual, expected)); in TEST_F()
269 ASSERT_TRUE(torch::allclose(actual, expected)); in TEST_F()
282 ASSERT_TRUE(torch::allclose(actual, expected)); in TEST_F()
295 ASSERT_TRUE(torch::allclose(actual, expected)); in TEST_F()
312 ASSERT_TRUE(torch::allclose(actual, expected)); in TEST_F()
329 ASSERT_TRUE(torch::allclose(actual, expected)); in TEST_F()
346 ASSERT_TRUE(torch::allclose(actual, expected)); in TEST_F()
[all …]
H A Dlegacy_vmap_test.cpp212 ASSERT_TRUE(at::allclose(result.tensor(), tensor.permute({1, 0, 2}))); in TEST()
222 ASSERT_TRUE(at::allclose(result.tensor(), tensor.permute({1, 2, 0}))); in TEST()
361 ASSERT_TRUE(at::allclose(out, x.sum(1))); in TEST()
371 ASSERT_TRUE(at::allclose(out, x.sum(0))); in TEST()
381 ASSERT_TRUE(at::allclose(out, x.sum(std::vector<int64_t>{0, 2}))); in TEST()
391 ASSERT_TRUE(at::allclose(out, x.sum(std::vector<int64_t>{2, 3}))); in TEST()
401 ASSERT_TRUE(at::allclose(output, expected_outputs[idx])); in checkBroadcastingVmapTransform()
585 ASSERT_TRUE(at::allclose(out, x * y)); in TEST()
597 ASSERT_TRUE(at::allclose(out, x * y)); in TEST()
612 ASSERT_TRUE(at::allclose(out, x.unsqueeze(1) * y)); in TEST()
[all …]
H A Dtest_assert.h52 // allclose broadcasts, so check same size before allclose.
55 ASSERT(t1.allclose(t2));
57 // allclose broadcasts, so check same size before allclose.
60 ASSERT(t1.allclose(t2, atol, rtol));
/aosp_15_r20/external/pytorch/test/custom_operator/
H A Dtest_custom_ops.cpp49 TORCH_INTERNAL_ASSERT(output[i].allclose(torch::ones(5) * 2)); in get_operator_from_registry_and_execute()
50 TORCH_INTERNAL_ASSERT(output[i].allclose(manual[i])); in get_operator_from_registry_and_execute()
62 TORCH_INTERNAL_ASSERT(output.allclose(x + 2*y + x*y)); in get_autograd_operator_from_registry_and_execute()
66 TORCH_INTERNAL_ASSERT(torch::allclose(x.grad(), y + torch::ones({5,5}))); in get_autograd_operator_from_registry_and_execute()
67 TORCH_INTERNAL_ASSERT(torch::allclose(y.grad(), x + torch::ones({5,5})*2)); in get_autograd_operator_from_registry_and_execute()
75 TORCH_INTERNAL_ASSERT(output.allclose(x + 2*y + x*y + z)); in get_autograd_operator_from_registry_and_execute()
79 TORCH_INTERNAL_ASSERT(torch::allclose(x.grad(), y + torch::ones({5,5}))); in get_autograd_operator_from_registry_and_execute()
80 TORCH_INTERNAL_ASSERT(torch::allclose(y.grad(), x + torch::ones({5,5})*2)); in get_autograd_operator_from_registry_and_execute()
81 TORCH_INTERNAL_ASSERT(torch::allclose(z.grad(), torch::ones({5,5}))); in get_autograd_operator_from_registry_and_execute()
93 TORCH_INTERNAL_ASSERT(output.allclose(x + 2*y + x*y)); in get_autograd_operator_from_registry_and_execute_in_nograd_mode()
[all …]
H A Dtest_custom_ops.py96 self.assertTrue(tensor.allclose(torch.ones(5) * 2))
101 self.assertTrue(output[0].allclose(torch.ones(5)))
107 self.assertTrue(output.allclose(x + 2 * y + x * y))
121 self.assertTrue(output.allclose(x + 2 * y + x * y + z))
134 self.assertTrue(output.allclose(x + 2 * y + x * y))
139 self.assertTrue(output.allclose(torch.ones(5) + 1))
155 self.assertTrue(output.allclose(torch.ones(5) + 1))
/aosp_15_r20/external/executorch/kernels/portable/
H A Dcustom_ops.yaml9 # This custom version of allclose is necessary because the native allclose's
14 # a single-element bool. The result of the allclose operation is written to that
18 - func: allclose.out(Tensor self, Tensor other, float rtol=1e-05, float atol=1e-08, bool equal_nan=…
24 # allclose(). Otherwise, code calling this op is identical to the native op:
25 # out = torch.ops.aten.allclose(a, b). The out variant pass attempts to convert
26 # this call to the nonexistent native allclose.out and fails.
29 - func: allclose.Tensor(Tensor self, Tensor other, float rtol=1e-05, float atol=1e-08, bool equal_n…
/aosp_15_r20/external/executorch/extension/llm/custom_ops/
H A Dtest_sdpa_with_kv_cache.py95 self.assertTrue(torch.allclose(ref_output, op_output))
137 self.assertTrue(torch.allclose(ref_output, op_output))
178 self.assertTrue(torch.allclose(ref_output, op_output))
219 self.assertTrue(torch.allclose(ref_output, op_output))
270 self.assertTrue(torch.allclose(ref_output, op_output))
288 self.assertTrue(torch.allclose(ref_output, op_output))
334 self.assertTrue(torch.allclose(ref_output, op_output))
350 self.assertTrue(torch.allclose(ref_output, op_output))
369 self.assertTrue(torch.allclose(ref_output, op_output))
449 self.assertTrue(torch.allclose(ref_output, op_output, atol=1e-6))
[all …]
/aosp_15_r20/external/pytorch/functorch/notebooks/
H A Daot_autograd_optimizations.ipynb133 "assert torch.allclose(ref, res)"
170 "assert torch.allclose(ref, res)\n",
171 "assert torch.allclose(a.grad, cloned_a.grad)\n",
172 "assert torch.allclose(b.grad, cloned_b.grad)\n",
173 "assert torch.allclose(c.grad, cloned_c.grad)\n",
174 "assert torch.allclose(d.grad, cloned_d.grad)"
339 "assert torch.allclose(ref, res)\n",
340 "assert torch.allclose(a.grad, cloned_a.grad)\n",
341 "assert torch.allclose(b.grad, cloned_b.grad)\n",
342 "assert torch.allclose(c.grad, cloned_c.grad)\n",
[all …]
/aosp_15_r20/external/pytorch/test/
H A Dtest_out_dtype_op.py36 self.assertTrue(torch.allclose(m(x), gm(x)))
39 self.assertTrue(torch.allclose(m(x), gm(x)))
42 self.assertTrue(torch.allclose(m(x), gm(x)))
69 self.assertTrue(torch.allclose(m(x), ep.module()(x)))
97 self.assertTrue(torch.allclose(numerical_res, gm(x)))
108 self.assertTrue(torch.allclose(f(*inp), compiled(*inp)))
120 self.assertTrue(torch.allclose(numerical_res, gm(*inp)))
179 self.assertTrue(torch.allclose(ref, test_out))
180 self.assertTrue(torch.allclose(ref, test_out_c))
/aosp_15_r20/external/executorch/exir/tests/
H A Dtest_tracer.py194 self.assertTrue(torch.allclose(orig_res[0], new_res[0]))
195 self.assertTrue(torch.allclose(orig_res[1], new_res[1]))
278 self.assertTrue(torch.allclose(Module()(*inp), gm(*inp)))
374 self.assertTrue(torch.allclose(graph_outputs[0], eager_outputs[0]))
375 self.assertTrue(torch.allclose(graph_outputs[1], eager_outputs[1]))
449 self.assertTrue(torch.allclose(ep(torch.ones(6, 4)), Foo()(torch.ones(6, 4))))
466 self.assertTrue(torch.allclose(ep(inp), FooContainerInputOutput()(inp)))
483 self.assertTrue(torch.allclose(ep(*inp), FooContainerInputOutputV2()(*inp)))
514 self.assertTrue(torch.allclose(ep(torch.ones(6, 4)), Foo()(torch.ones(6, 4))))
554 torch.allclose(
[all …]
/aosp_15_r20/external/pytorch/test/distributed/_shard/sharded_tensor/ops/
H A Dtest_binary_cmp.py27 …"""Test base for binary comparison functions such as torch.equal, torch.allclose etc. for ShardedT…
140 self._test_common_failures(torch.allclose)
146 """Test torch.allclose(ShardedTensor, ShardedTensor)"""
151 self.assertTrue(torch.allclose(st1, st2))
152 self.assertTrue(torch.allclose(st1, st2, atol=0))
156 self.assertFalse(torch.allclose(st1, st2))
158 self.assertTrue(torch.allclose(st1, st2, atol=1))
/aosp_15_r20/external/pytorch/torch/_functorch/
H A Deager_transforms.py275 >>> assert torch.allclose(grad, torch.func.grad(f)(x))
284 >>> assert torch.allclose(vjps[0], x.cos() + -x.sin())
293 >>> assert torch.allclose(vjps[0], x.cos() + -x.sin())
303 >>> assert torch.allclose(vjps[0], torch.matmul(cotangents, y.transpose(0, 1)))
304 >>> assert torch.allclose(vjps[1], torch.matmul(x.transpose(0, 1), cotangents))
315 >>> assert torch.allclose(vjps[0], torch.full(x.shape, 4.))
514 >>> assert torch.allclose(jacobian, expected)
531 >>> assert torch.allclose(f_x, f(x))
550 >>> assert torch.allclose(hessian, torch.diag(-x.sin()))
563 >>> assert torch.allclose(jacobian, expected)
[all …]
/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_allclose.cpp120 * Note: This custom operator contains two variants: allclose.Tensor (a
121 * functional variant, no inplace mutating on the arguments) and allclose.out
124 * only allclose.out will be seen from ExecuTorch runtime. With this setup, the
125 * portable kernel for allclose.Tensor can be implemented as a wrapper of
126 * allclose.out. We can easily instantiate an at::Tensor for the out argument,
127 * then pass it into allclose.out. This logic will only need to work out in
128 * "ATen mode" for ExecuTorch compiler, since we won't expose allclose.Tensor in
/aosp_15_r20/external/executorch/test/end2end/
H A Dtest_end2end.py412 def allclose(lhs, rhs, rtol=1e-5, atol=1e-8): function
414 Unlike torch.allocse which only handles Tensor arguments, allclose handles
418 return torch.allclose(lhs, rhs, rtol, atol)
421 allclose(a, b, rtol, atol) for a, b in zip(lhs, rhs)
428 return all(allclose(lhs[k], rhs[k], rtol, atol) for k in lhs)
493 atol: Absolute tolerance used in allclose and torch.allclose
494 rtol: Relative tolerance used in allclose and torch.allclose
530 self.assertTrue(allclose(expected, result, rtol, atol))
568 is_close = allclose(expected, actual, rtol, atol)

12345678910>>...14