Searched refs:mean4D (Results 1 – 1 of 1) sorted by relevance
502 Array4D<float> mean4D = in XLA_TEST_P() local509 auto normalized = *ReferenceUtil::BatchNorm4D(input_array, mean4D, var4D, in XLA_TEST_P()603 Array4D<float> mean4D = in XLA_TEST_P() local610 auto normalized = *ReferenceUtil::BatchNorm4D(input_array, mean4D, var4D, in XLA_TEST_P()722 Array4D<float> mean4D = in XLA_TEST_P() local738 input_array, mean4D, [](float a, float b) { return a - b; }); in XLA_TEST_P()