Home
last modified time | relevance | path

Searched refs:CheckNumericsLaunchV2 (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcheck_numerics_op_gpu.cu.cc103 struct CheckNumericsLaunchV2 { struct
117 template struct CheckNumericsLaunchV2<Eigen::half>; variable
118 template struct CheckNumericsLaunchV2<float>; variable
119 template struct CheckNumericsLaunchV2<double>; variable
Dcheck_numerics_op.cc58 struct CheckNumericsLaunchV2 { struct
63 extern template struct CheckNumericsLaunchV2<Eigen::half>;
64 extern template struct CheckNumericsLaunchV2<float>;
65 extern template struct CheckNumericsLaunchV2<double>;
322 CheckNumericsLaunchV2<T>().Run(d, data, size, abnormal_detected); in RunKernel()