Function TestCase1

tensorflow/core/kernels/data/map_defun_op_test.cc:80–94

Test case 1: one input for the map function with no captured inputs.

// Test case 1: one input for the map function with no captured inputs.
TestCase TestCase1() {
  return {
      /*arguments*/ {
          CreateTensor<int64>(TensorShape({3, 2}), {0, 1, 2, 3, 4, 5})},
      /*captured_inputs*/ {},
      /*t_arguments*/ {DT_INT64},
      /*t_captured*/ {},
      /*func*/ {FunctionDefHelper::FunctionRef("XTimesTwo", {{"T", DT_INT64}})},
      /*func_lib*/ {test::function::XTimesTwo()},
      /*max_intra_op_parallelism*/ 2,
      /*output_dtypes*/ {DT_INT64},
      /*output_shapes*/ {PartialTensorShape({2})},
      /*expected_outputs*/
      {CreateTensor<int64>(TensorShape({3, 2}), {0, 2, 4, 6, 8, 10})}};
}

// Test case 2: two inputs for the map function with no captured inputs.
TestCase TestCase2() {