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

Function TestCase2

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

Test case 2: two inputs for the map function with no captured inputs.

Source from the content-addressed store, hash-verified

95
96// Test case 2: two inputs for the map function with no captured inputs.
97TestCase TestCase2() {
98 return {
99 /*arguments*/ {
100 CreateTensor<int64>(TensorShape({3, 2}), {0, 1, 2, 3, 4, 5}),
101 CreateTensor<int64>(TensorShape({3, 2}), {0, 10, 20, 30, 40, 50})},
102 /*captured_inputs*/ {},
103 /*t_arguments*/ {DT_INT64, DT_INT64},
104 /*t_captured*/ {},
105 /*func*/ {FunctionDefHelper::FunctionRef("XAddY", {{"T", DT_INT64}})},
106 /*func_lib*/ {test::function::XAddY()},
107 /*max_intra_op_parallelism*/ 2,
108 /*output_dtypes*/ {DT_INT64},
109 /*output_shapes*/ {PartialTensorShape({2})},
110 /*expected_outputs*/
111 {CreateTensor<int64>(TensorShape({3, 2}), {0, 11, 22, 33, 44, 55})}};
112}
113
114// Test case 3: two inputs for the map function with one captured input.
115TestCase TestCase3() {

Callers 1

Calls 3

XAddYFunction · 0.85
PartialTensorShapeClass · 0.85
TensorShapeClass · 0.50

Tested by

no test coverage detected