Searched refs:input_array_ (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dbatch_normalization_test.cc55 BatchNormalizationTest() : input_array_(kSamples, kZ, kY, kX) {
64 input_array_.FillWithPZ(pz);
65 input_literal_ = std::move(*Literal::CreateR4FromArray4D(input_array_));
66 CHECK_EQ(kSamples, input_array_.planes());
67 CHECK_EQ(kZ, input_array_.depth());
68 CHECK_EQ(kY, input_array_.height());
69 CHECK_EQ(kY, input_array_.width());
77 Array4D<float> input_array_; member in class:xla::__anon25717::BatchNormalizationTest

Completed in 135 milliseconds