Searched refs:temp_stats_double (Results 1 – 1 of 1) sorted by relevance
243 auto temp_stats_double = temp_stats_double_t.tensor<double, 4>(); in Compute() local244 temp_stats_double.setZero(); in Compute()252 temp_stats_double(feature_idx, node, bucket, 0) += gradients(i, 0); in Compute()253 temp_stats_double(feature_idx, node, bucket, 1) += hessians(i, 0); in Compute()263 temp_stats_double.template cast<float>(); in Compute()