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