Searched refs:DoWithUpcastToF32 (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | math.cc | 65 static XlaOp DoWithUpcastToF32(XlaOp operand, in DoWithUpcastToF32() function
292 return DoWithUpcastToF32(x, {BF16, F16}, [](XlaOp x) { in Erfc()
333 return DoWithUpcastToF32(x, {BF16, F16}, in Erf()
482 return DoWithUpcastToF32(x, {BF16, F16}, in ErfInv()
611 return DoWithUpcastToF32(input, {BF16, F16}, do_it); in Lgamma()
706 return DoWithUpcastToF32(input, {BF16, F16}, do_it); in Digamma()
1169 return DoWithUpcastToF32(x, {F16}, [](XlaOp x) { return Sin(x) / Cos(x); }); in Tan()
1260 return DoWithUpcastToF32(x, {BF16, F16}, [&](XlaOp x) { in Asinh()
1282 return DoWithUpcastToF32(x, {BF16}, [&](XlaOp x) { // in Atanh()
1298 return DoWithUpcastToF32(x, {BF16, F16}, [](XlaOp x) { in Cosh()
[all …]
|