Searched refs:num_elements_per_feature (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dbatch_normalization_test.cc449 const int64 num_elements_per_feature = local
475 mean[i] = sum[i] / num_elements_per_feature;
485 square_mean[i] = sum_squared[i] / num_elements_per_feature;
549 const int64 num_elements_per_feature = local
575 mean[i] = sum[i] / num_elements_per_feature;
585 square_mean[i] = sum_squared[i] / num_elements_per_feature;
661 const int64 num_elements_per_feature = local
686 if (num_elements_per_feature > 0) {
687 mean[i] = sum[i] / num_elements_per_feature;
700 if (num_elements_per_feature >
[all...]

Completed in 98 milliseconds