Function XLA_TEST_F

tensorflow/compiler/xla/tests/dot_operation_test.cc:68–85

Check that we can safely pass an input tuple's elements to a dot operation.

// Check that we can safely pass an input tuple's elements to a dot operation.
XLA_TEST_F(DotOperationTest, DotOfInputTupleElem) {
  XlaBuilder builder(TestName());

  XlaOp param;
  auto param_data = CreateParameterAndTransferLiteral(
      0,
      LiteralUtil::MakeTupleFromSlices(
          {LiteralUtil::CreateR2<float>({{1, 2}, {3, 4}}),
           LiteralUtil::CreateR2<float>({{5, 6}, {7, 8}})}),
      "arg0", &builder, &param);
  auto lhs = GetTupleElement(param, 0);
  auto rhs = GetTupleElement(param, 1);
  Dot(lhs, rhs);

  ComputeAndCompareLiteral(&builder,
                           LiteralUtil::CreateR2<float>({{19, 22}, {43, 50}}),
                           {param_data.get()});
}

template <typename T>
class DotOperationTest_F16F32F64CF64 : public DotOperationTest {};