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

Function TestCase3

tensorflow/core/kernels/data/map_defun_op_test.cc:115–130  ·  view source on GitHub ↗

Test case 3: two inputs for the map function with one captured input.

Source from the content-addressed store, hash-verified

113
114// Test case 3: two inputs for the map function with one captured input.
115TestCase TestCase3() {
116 return {
117 /*arguments*/ {
118 CreateTensor<int64>(TensorShape({3, 2}), {0, 1, 2, 3, 4, 5})},
119 /*captured_inputs*/
120 {CreateTensor<int64>(TensorShape({2}), {10, 100})},
121 /*t_arguments*/ {DT_INT64},
122 /*t_captured*/ {DT_INT64},
123 /*func*/ {FunctionDefHelper::FunctionRef("XAddY", {{"T", DT_INT64}})},
124 /*func_lib*/ {test::function::XAddY()},
125 /*max_intra_op_parallelism*/ 2,
126 /*output_dtypes*/ {DT_INT64},
127 /*output_shapes*/ {PartialTensorShape({2})},
128 /*expected_outputs*/
129 {CreateTensor<int64>(TensorShape({3, 2}), {10, 101, 12, 103, 14, 105})}};
130}
131
132TestCase InvalidOutputTypes() {
133 return {

Callers 1

Calls 3

XAddYFunction · 0.85
PartialTensorShapeClass · 0.85
TensorShapeClass · 0.50

Tested by

no test coverage detected