Searched refs:ErfInv (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | exhaustive_unary_test_f32_or_smaller.cc | 593 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 __anon9015410b1d02()
599 UNARY_TEST_BF16(ErfInv, { Run(ErfInv, HostErfInv); })
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | unary_ops.cc | 126 XLAJIT_MAKE_UNARY(Erfinv, xla::ErfInv(x));
129 xla::ErfInv(xla::ScalarLike(x, 2.0) * x -
|
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| D | random.cc | 75 xla::XlaOp x = mu + sigma * sqrt_2 * xla::ErfInv(v); in ParameterizedTruncatedNormal()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | math.h | 53 XlaOp ErfInv(XlaOp x);
|
| D | math_test.cc | 151 ErfInv(x); in TestErfEdgeCases()
258 ErfInv(x); in XLA_TEST_F()
|
| D | math.cc | 474 XlaOp ErfInv(XlaOp x) { in ErfInv() function
|
| /external/pytorch/aten/src/ATen/cpu/ |
| D | vml.h | 147 IMPLEMENT_VML_MKL(erfinv, ErfInv)
|
| /external/tensorflow/tensorflow/compiler/xla/python/xla_extension/ |
| D | ops.pyi | 445 def ErfInv(__arg: XlaOp) -> XlaOp: ...
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | ops.cc | 489 UNARY_OP(ErfInv); in BuildOpsSubmodule()
|