Searched refs:num_elements_per_feature (Results 1 – 1 of 1) sorted by relevance
458 const int64 num_elements_per_feature = in XLA_TEST_P() local484 mean[i] = sum[i] / num_elements_per_feature; in XLA_TEST_P()494 square_mean[i] = sum_squared[i] / num_elements_per_feature; in XLA_TEST_P()559 const int64 num_elements_per_feature = in XLA_TEST_P() local585 mean[i] = sum[i] / num_elements_per_feature; in XLA_TEST_P()595 square_mean[i] = sum_squared[i] / num_elements_per_feature; in XLA_TEST_P()671 const int64 num_elements_per_feature = in XLA_TEST_P() local696 if (num_elements_per_feature > 0) { in XLA_TEST_P()697 mean[i] = sum[i] / num_elements_per_feature; in XLA_TEST_P()710 if (num_elements_per_feature > 0) { in XLA_TEST_P()[all …]