Home
last modified time | relevance | path

Searched refs:ref_input (Results 1 – 11 of 11) sorted by relevance

/external/libaom/test/
Dav1_fwd_txfm1d_test.cc73 std::unique_ptr<double[]> ref_input(new (std::nothrow) double[txfm_size]); in TEST() local
77 ASSERT_NE(ref_input, nullptr); in TEST()
90 ref_input[ni] = static_cast<double>(input[ni]); in TEST()
94 reference_hybrid_1d(ref_input.get(), ref_output.get(), txfm_size, in TEST()
Dav1_inv_txfm2d_test.cc87 double ref_input[64 * 64]; in RunRoundtripCheck() local
88 ASSERT_LE(txfm2d_size, NELEMENTS(ref_input)); in RunRoundtripCheck()
90 ref_input[ni] = input[ni]; in RunRoundtripCheck()
95 libaom_test::reference_hybrid_2d(ref_input, ref_coeffs, tx_type_, in RunRoundtripCheck()
/external/pytorch/test/nn/
Dtest_pooling.py227 ref_input = input.detach().clone().contiguous().requires_grad_(True)
233 ref_out = ref_pool(ref_input)
239 self.assertEqual(input.grad, ref_input.grad)
254 ref_input = input.detach().clone().contiguous().requires_grad_(True)
260 ref_out = ref_pool(ref_input)
266 self.assertEqual(input.grad, ref_input.grad)
324 ref_input = input.detach().clone().contiguous().requires_grad_(True)
330 ref_out = ref_pool(ref_input)
336 self.assertEqual(input.grad, ref_input.grad)
347 ref_input = input.detach().clone().contiguous().requires_grad_(True)
[all …]
Dtest_convolution.py3425 ref_input = input.detach().clone().contiguous().requires_grad_()
3436 ref_out = ref_conv(ref_input)
3449 self.assertEqual(input.grad, ref_input.grad, exact_dtype=False)
3626 ref_input = input.detach().clone().contiguous().double().requires_grad_()
3635 ref_out = ref_conv(ref_input)
3650 self.assertTrue(ref_input.grad.is_contiguous())
3656 self.assertEqual(input.grad, ref_input.grad, exact_dtype=False)
3680 ref_input = input.detach().clone().contiguous().double().requires_grad_()
3689 ref_out = ref_conv(ref_input)
3706 self.assertTrue(ref_input.grad.is_contiguous())
[all …]
/external/tensorflow/tensorflow/python/keras/engine/
Dfunctional.py537 y = self._conform_to_reference_input(y, ref_input=x)
603 def _conform_to_reference_input(self, tensor, ref_input): argument
610 ref_shape = ref_input.shape
629 tensor.set_shape(tensor.shape.merge_with(ref_input.shape))
634 ref_input.shape, ref_input, tensor.shape))
637 tensor = math_ops.cast(tensor, dtype=ref_input.dtype)
641 ref_input_dtype = getattr(ref_input, 'dtype', None)
/external/pytorch/test/xpu/
Dtest_conv.py1010 ref_input = input.detach().clone().contiguous().double().requires_grad_()
1018 ref_out = ref_conv(ref_input)
1033 self.assertTrue(ref_input.grad.is_contiguous())
1039 self.assertEqual(input.grad, ref_input.grad, exact_dtype=False)
1059 ref_input = input.detach().clone().contiguous().double().requires_grad_()
1067 ref_out = ref_conv(ref_input)
1084 self.assertTrue(ref_input.grad.is_contiguous())
1090 self.assertEqual(input.grad, ref_input.grad, exact_dtype=False)
1104 ref_input, argument
1132 self.assertEqual(input.grad, ref_input.grad)
[all …]
/external/pytorch/torchgen/api/
Dautograd.py430 for ref_input in derivative.saved_inputs:
431 ref_input_jit_name = ref_input.expr.split(".")[0]
437 new_expr = ref_input.expr.replace(ref_input_jit_name, mapped_expr)
439 cast(str, ref_input.nctype.name), new_expr
/external/pytorch/test/
Dtest_foreach.py220 ref_input, ctxmgr = sample.input, nullcontext()
223 ref_input = [t.clone().detach() for t in sample.input]
235 ref([ref_input, *sample.ref_args], **ref_kwargs)
237 expected = ref([ref_input, *sample.ref_args], **ref_kwargs)
1294 ref_input = [t.clone().detach() for t in sample.input]
1296 for t, s in zip(ref_input, sample.args[0]):
1298 self.assertEqual(sample.input, ref_input)
1303 for t, s in zip(ref_input, rhs_tensors):
1305 self.assertEqual(ref_input, sample.input)
Dtest_nn.py4507 ref_input = input.detach().clone().contiguous().requires_grad_(True)
4514 ref_out = ref_pus(ref_ps(ref_input))
4520 self.assertEqual(input.grad, ref_input.grad)
4796 ref_input = input.detach().clone().contiguous().requires_grad_(True)
4807 ref_out = ref_bn(ref_input)
4815 self.assertEqual(input.grad, ref_input.grad)
4849 ref_input = input.detach().clone().contiguous()
4854 ref_out = ref_bn(ref_input)
4880 ref_input = input.detach().clone().contiguous().requires_grad_(True)
4886 ref_out = ref_mod(ref_input)
[all …]
/external/tensorflow/tensorflow/core/tfrt/utils/
Dtfrt_graph_execution_state.cc714 std::string ref_input; in EliminateRefVariablesFromV1ControlFlow() local
718 ref_input = std::move(input); in EliminateRefVariablesFromV1ControlFlow()
722 if (!ref_input.empty()) { in EliminateRefVariablesFromV1ControlFlow()
729 "Cannot in-place update ref node ", ref_input, in EliminateRefVariablesFromV1ControlFlow()
/external/pytorch/torch/testing/_internal/
Dcommon_nn.py3426 ref_input = deepcopy(input)
3428 … expected_out = self.reference_fn(ref_input, test_case._get_parameters(module)[0], ref_module)