Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
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
}
51
template <>
52
__device__ Eigen::half impl_sqrt(Eigen::half x) {
53
return __float2half(sqrt(__half2float(x)));
54
}
55
template <>
56
__device__ Eigen::half impl_rsqrt(Eigen::half x) {
57
return __float2half(rsqrt(__half2float(x)));
Callers
nothing calls this directly
Calls
1
sqrt
Class · 0.70
Tested by
no test coverage detected