Home
last modified time | relevance | path

Searched refs:numpy_fn (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dnccl_ops_test.py59 numpy_fn, argument
85 np_ans = numpy_fn(np_ans, t)
105 def _TestGradient(self, nccl_reduce, numpy_fn): argument
124 self._Test(_Gradient, numpy_fn)
/aosp_15_r20/external/pytorch/docs/source/
H A Dtorch.compiler_faq.rst474 def numpy_fn(X: np.ndarray, Y: np.ndarray) -> np.ndarray:
479 Z = numpy_fn(X, Y)
502 def numpy_fn(X: np.ndarray, Y: np.ndarray) -> np.ndarray:
508 Z = numpy_fn(X, Y)
511 In this example, ``numpy_fn`` will be executed in CUDA. For this to be
516 to tweak our ``numpy_fn`` so that it accepts cuda Tensors and returns tensors.
523 def numpy_fn(X, Y):
528 Z = numpy_fn(X, Y)
545 def numpy_fn(X, Y):
550 Z = numpy_fn(X, Y)
[all …]
/aosp_15_r20/external/pytorch/test/
H A Dtest_spectral_ops.py595 numpy_fn = getattr(scipy.fft, fname)
597 numpy_fn = getattr(np.fft, fname)
606 expected = numpy_fn(input_np, s, norm=norm)
613 expected = numpy_fn(input_np, s, dim, norm)
699 numpy_fn = getattr(np.fft, fname)
703 expected = numpy_fn(*args)
739 numpy_fn = getattr(np.fft, fname)
741 expected = numpy_fn(input_np, axes=dim)
H A Dtest_tensor_creation_ops.py2650 def test_fn(torch_fn, numpy_fn, steps): argument
2652 a = numpy_fn(start, end, steps, dtype=torch_to_numpy_dtype_dict[dtype])
/aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/definitions/
H A D_masked.py238 numpy_fn, argument
240 ref = reference_reduction_numpy(numpy_fn)