Function TestCase1

tensorflow/core/kernels/data/shard_dataset_op_test.cc:70–84

Test Case 1: simple case.

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

// Test Case 2: zero offset.
TestCase TestCase2() {