MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / half impl_sqrt

Method half impl_sqrt

tensorflow/core/kernels/training_ops_gpu.cu.cc:52–54  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

50}
51template <>
52__device__ Eigen::half impl_sqrt(Eigen::half x) {
53 return __float2half(sqrt(__half2float(x)));
54}
55template <>
56__device__ Eigen::half impl_rsqrt(Eigen::half x) {
57 return __float2half(rsqrt(__half2float(x)));

Callers

nothing calls this directly

Calls 1

sqrtClass · 0.70

Tested by

no test coverage detected