Searched refs:BatchNorm4D (Results 1 – 3 of 3) sorted by relevance
607 auto normalized = *ReferenceUtil::BatchNorm4D(input_array, mean4D, var4D, in XLA_TEST_P()708 auto normalized = *ReferenceUtil::BatchNorm4D(input_array, mean4D, var4D, in XLA_TEST_P()
209 static std::unique_ptr<Array4D<float>> BatchNorm4D(
308 /* static */ std::unique_ptr<Array4D<float>> ReferenceUtil::BatchNorm4D( in BatchNorm4D() function in xla::ReferenceUtil