Home
last modified time | relevance | path

Searched defs:square_mean (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dbatch_normalization_test.cc555 std::vector<float> square_mean(feature_bound); in XLA_TEST_P() local
656 std::vector<float> square_mean(feature_bound); in XLA_TEST_P() local
771 std::vector<float> square_mean(feature_bound); in XLA_TEST_P() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dbatchnorm_expander.cc231 auto square_mean = add(Mean(elements_per_feature, squared_sum, add)); in HandleBatchNormTraining() local