Home
last modified time | relevance | path

Searched defs:reference_fn (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_modules.py165 def __init__(self, constructor_input, forward_input=None, desc='', reference_fn=None): argument
330 def reference_fn(m, p, i, t, constructor_kwargs=constructor_kwargs): function
377 def reference_fn(m, p, i, t, constructor_kwargs=constructor_kwargs): function
478 def reference_fn(m, p, i, t, constructor_kwargs=constructor_kwargs): function
990 def reference_fn(m, p, i1, i2, t, constructor_kwargs=constructor_kwargs): function
1428 def reference_fn(m, p, i, t, constructor_kwargs=constructor_kwargs): function
1556 def reference_fn(m, p, i, t, reduction=reduction, constructor_kwargs=constructor_kwargs): function
1644 def reference_fn(m, p, i, t, il, tl, constructor_kwargs=constructor_kwargs): function
1809 def reference_fn(m, p, i, t, constructor_kwargs=constructor_kwargs): function
1842 def reference_fn(m, p, i, t, constructor_kwargs=constructor_kwargs): function
[all …]
H A Dcommon_nn.py3326 def __init__(self, constructor, desc='', reference_fn=None, fullname=None, **kwargs): argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/python_tests/
H A Dtf_logical_ops_test.py41 def test_logical_op(mlir_blob, reference_fn, rank): argument
/aosp_15_r20/external/pytorch/test/
H A Dtest_binary_ufuncs.py4230 def test_helper(torch_fn, reference_fn, inputs, scalar=None): argument
4282 def test_zeros_special_helper(torch_fn, reference_fn, scalar=False): argument
4333 def _compare_helper(x, y, torch_fn, reference_fn): argument
H A Dtest_namedtensor.py1715 def reference_fn(*tensors): function
H A Dtest_nn.py7332 def reference_fn(i, p, m): function