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