Function XLA_TEST_F

tensorflow/compiler/xla/tests/convolution_variants_test.cc:54–67

};

XLA_TEST_F(ConvolutionVariantsTest, Minimal) {
  XlaBuilder builder(TestName());

  const Array4D<float> input_array(1, 1, 1, 1, {2});
  auto input = ConstantR4FromArray4D<float>(&builder, input_array);

  const Array4D<float> filter_array(1, 1, 1, 1, {3});
  auto filter = ConstantR4FromArray4D<float>(&builder, filter_array);

  Conv(input, filter, {1, 1}, Padding::kValid);

  const Array4D<float> expected(1, 1, 1, 1, {6});
  ComputeAndCompareR4<float>(&builder, expected, {}, error_spec_);
}

XLA_TEST_F(ConvolutionVariantsTest, MinimalWithBatch) {
  XlaBuilder builder(TestName());