HomeSort by relevance Sort by last modified time
    Searched refs:CheckNumericsLaunch (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
check_numerics_op_gpu.cu.cc 60 struct CheckNumericsLaunch {
73 template struct CheckNumericsLaunch<Eigen::half>;
74 template struct CheckNumericsLaunch<float>;
75 template struct CheckNumericsLaunch<double>;
check_numerics_op.cc 38 struct CheckNumericsLaunch {
142 CheckNumericsLaunch<T>().Run(d, input.data(), input.size(),

Completed in 109 milliseconds