Function TestCase4

tensorflow/core/kernels/data/shard_dataset_op_test.cc:119–132

Test Case 4: larger num_shards.

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

// Test Case 5: index == num_shards.
TestCase TestCase5() {