Home
last modified time | relevance | path

Searched refs:ErfInv (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dexhaustive_unary_test_f32_or_smaller.cc593 UNARY_TEST_F32(ErfInv, { Run(ErfInv, HostErfInv); })
595 UNARY_TEST_F16(ErfInv, {
596 Run(ErfInv, HostErfInv, [](Eigen::half) { return ErrorSpec{0.002, 0.002}; }); in __anon868b568b1d02()
599 UNARY_TEST_BF16(ErfInv, { Run(ErfInv, HostErfInv); })
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dunary_ops.cc126 XLAJIT_MAKE_UNARY(Erfinv, xla::ErfInv(x));
129 xla::ErfInv(xla::ScalarLike(x, 2.0) * x -
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Dmath.h53 XlaOp ErfInv(XlaOp x);
H A Dmath_test.cc151 ErfInv(x); in TestErfEdgeCases()
258 ErfInv(x); in XLA_TEST_F()
H A Dmath.cc474 XlaOp ErfInv(XlaOp x) { in ErfInv() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/lib/
H A Drandom.cc75 xla::XlaOp x = mu + sigma * sqrt_2 * xla::ErfInv(v); in ParameterizedTruncatedNormal()
/aosp_15_r20/external/pytorch/aten/src/ATen/cpu/
H A Dvml.h147 IMPLEMENT_VML_MKL(erfinv, ErfInv)
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/xla_extension/
H A Dops.pyi445 def ErfInv(__arg: XlaOp) -> XlaOp: ...
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dops.cc489 UNARY_OP(ErfInv); in BuildOpsSubmodule()