Searched refs:input_array_ (Results 1 – 1 of 1) sorted by relevance
56 BatchNormalizationTest() : input_array_(kSamples, kZ, kY, kX) { in BatchNormalizationTest()65 input_array_.FillWithPZ(pz); in BatchNormalizationTest()66 input_literal_ = LiteralUtil::CreateR4FromArray4D(input_array_); in BatchNormalizationTest()67 CHECK_EQ(kSamples, input_array_.planes()); in BatchNormalizationTest()68 CHECK_EQ(kZ, input_array_.depth()); in BatchNormalizationTest()69 CHECK_EQ(kY, input_array_.height()); in BatchNormalizationTest()70 CHECK_EQ(kY, input_array_.width()); in BatchNormalizationTest()87 Array4D<float> input_array_; member in xla::__anoncfc9dfd00111::BatchNormalizationTest