Searched refs:rsqrt_cost (Results 1 – 1 of 1) sorted by relevance
2381 const auto rsqrt_cost = Eigen::internal::functor_traits< in PredictFusedBatchNorm() local2384 ops = dims.iz * (dims.batch * dims.ix * dims.iy * 4 + 6 + rsqrt_cost); in PredictFusedBatchNorm()2430 const auto rsqrt_cost = Eigen::internal::functor_traits< in PredictFusedBatchNormGrad() local2432 ops = dims.iz * (dims.batch * dims.ix * dims.iy * 11 + 5 + rsqrt_cost); in PredictFusedBatchNormGrad()