Home
last modified time | relevance | path

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

12345678910>>...13

/external/pytorch/test/jit/
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 …]
/external/pytorch/test/functorch/
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 …]
/external/pytorch/test/cpp/api/
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 …]
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 …]
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 …]
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 …]
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 …]
Divalue.cpp53 ASSERT_TRUE(copied_list[0].get().toTensor().allclose( in TEST()
55 ASSERT_TRUE(copied_list[1].get().toTensor().allclose( in TEST()
57 ASSERT_TRUE(copied_list[2].get().toTensor().allclose( in TEST()
59 ASSERT_TRUE(copied_list[3].get().toTensor().allclose( in TEST()
61 ASSERT_TRUE(copied_list[4].get().toTensor().allclose( in TEST()
Doperations.cpp26 ASSERT_TRUE(out.allclose(scalar_expected)); in TEST_F()
35 ASSERT_TRUE(out.allclose(tensor_expected)); in TEST_F()
58 ASSERT_TRUE(out.allclose(exp)); in TEST_F()
74 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
88 ASSERT_TRUE(torch::allclose(y, y_exp)); in TEST_F()
/external/executorch/kernels/portable/
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…
/external/pytorch/aten/src/ATen/test/
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));
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 …]
Doperators_test.cpp21 ASSERT_TRUE(at::allclose(result, a * b)); in TEST()
34 ASSERT_TRUE(at::allclose(result, expected)); in TEST()
43 ASSERT_TRUE(at::allclose(result, a.sin())); in TEST()
53 ASSERT_TRUE(at::allclose(result, a.sin())); in TEST()
/external/pytorch/test/cpp/lazy/
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 …]
/external/pytorch/test/custom_operator/
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 …]
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))
/external/pytorch/test/cpp/aoti_inference/
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 …]
/external/pytorch/test/distributed/_shard/sharded_tensor/ops/
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))
/external/executorch/kernels/portable/cpu/
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
/external/pytorch/test/
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))
/external/pytorch/test/cpp/tensorexpr/
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 …]
/external/executorch/exir/tests/
Dtest_joint_graph.py107 self.assertTrue(torch.allclose(loss, et_outputs[0]))
109 torch.allclose(m.linear.weight.grad, et_outputs[1]) # pyre-ignore[6]
111 self.assertTrue(torch.allclose(m.linear.bias.grad, et_outputs[2]))
112 self.assertTrue(torch.allclose(m.linear.weight, et_outputs[3]))
113 self.assertTrue(torch.allclose(m.linear.bias, et_outputs[4]))
/external/executorch/exir/dialects/backend/test/
Dtest_backend_ops.py52 self.assertTrue(self.torch_foo(a).allclose(self.backend_foo(a)))
53 self.assertTrue(self.edge_foo(a).allclose(self.backend_foo(a)))
76 self.assertTrue(op(a).allclose(f(a)))
88 self.assertTrue(op(a).allclose(f(a)))
/external/executorch/runtime/test/
Dtest_runtime.py31 self.assertTrue(torch.allclose(outputs[0], inputs[0] + inputs[1]))
41 self.assertTrue(torch.allclose(outputs[0], inputs[0] + inputs[1]))
45 self.assertTrue(torch.allclose(outputs[0], inputs[0] + inputs[1] + 1))
63 self.assertTrue(torch.allclose(outputs[0], inputs[0] + inputs[1]))
/external/executorch/extension/llm/custom_ops/
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 …]

12345678910>>...13