MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / TestCase3

Function TestCase3

tensorflow/core/kernels/data/reduce_dataset_op_test.cc:123–138  ·  view source on GitHub ↗

Test case 3: the input dataset has no outputs, so the reduce dataset just returns the initial state.

Source from the content-addressed store, hash-verified

121// Test case 3: the input dataset has no outputs, so the reduce dataset just
122// returns the initial state.
123TestCase TestCase3() {
124 return {/*range_data_param*/ {0, 0, 1},
125 /*initial_state*/
126 {CreateTensor<int64>(TensorShape({}), {1}),
127 CreateTensor<int64>(TensorShape({}), {3})},
128 /*func*/
129 FunctionDefHelper::FunctionRef("XAddY", {{"T", DT_INT64}}),
130 /*func_lib*/ {test::function::XAddY()},
131 /*t_state*/ {DT_INT64, DT_INT64},
132 /*use_inter_op_parallelism*/ true,
133 /*expected_outputs*/
134 {CreateTensor<int64>(TensorShape({}), {1}),
135 CreateTensor<int64>(TensorShape({}), {3})},
136 /*output_dtypes*/ {DT_INT64, DT_INT64},
137 /*output_shapes*/ {PartialTensorShape({}), PartialTensorShape({})}};
138}
139
140class ParameterizedReduceDatasetOpTest
141 : public ReduceDatasetOpTest,

Callers 1

Calls 3

XAddYFunction · 0.85
PartialTensorShapeClass · 0.85
TensorShapeClass · 0.50

Tested by

no test coverage detected