Function TestCase2

tensorflow/core/kernels/data/shard_dataset_op_test.cc:87–101

Test Case 2: zero offset.

// Test Case 2: zero offset.
TestCase TestCase2() {
  return {/*range_data_param*/ {0, 10, 1},
          /*num_shards*/
          CreateTensor<int64>(TensorShape({}), {5}),
          /*index*/
          CreateTensor<int64>(TensorShape({}), {0}),
          /*require_non_empty*/ true,
          /*expected_outputs*/
          {CreateTensor<int64>(TensorShape({}), {0}),
           CreateTensor<int64>(TensorShape({}), {5})},
          /*expected_output_dtypes*/ {DT_INT64},
          /*expected_output_shapes*/ {PartialTensorShape({})},
          /*expected_cardinality*/ 2,
          /*breakpoints*/ {0, 1, 5}};
}

// Test Case 3: iterator ends before first element.
TestCase TestCase3() {