/external/tensorflow/tensorflow/contrib/lite/kernels/ |
D | strided_slice_test.cc | 61 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput() function in tflite::__anon5c572b230111::StridedSliceOpModel 93 EXPECT_THAT(m.GetOutput(), ElementsAreArray({2, 3})); in TEST() 114 EXPECT_THAT(m.GetOutput(), ElementsAreArray({2, 3})); in TEST() 125 EXPECT_THAT(m.GetOutput(), ElementsAreArray({1, 2, 3})); in TEST() 136 EXPECT_THAT(m.GetOutput(), ElementsAreArray({2})); in TEST() 147 EXPECT_THAT(m.GetOutput(), ElementsAreArray({2, 3, 4})); in TEST() 158 EXPECT_THAT(m.GetOutput(), ElementsAreArray({1, 2, 3})); in TEST() 169 EXPECT_THAT(m.GetOutput(), ElementsAreArray({3})); in TEST() 180 EXPECT_THAT(m.GetOutput(), ElementsAreArray({4})); in TEST() 191 EXPECT_THAT(m.GetOutput(), ElementsAreArray({3, 2})); in TEST() [all …]
|
D | skip_gram_test.cc | 48 std::vector<string> GetOutput() { in GetOutput() function in tflite::__anon3bbaeba20111::SkipGramOp 70 EXPECT_THAT(m.GetOutput(), testing::UnorderedElementsAreArray( in TEST() 79 EXPECT_THAT(m.GetOutput(), in TEST() 89 EXPECT_THAT(m.GetOutput(), in TEST() 103 EXPECT_THAT(m.GetOutput(), in TEST() 122 m.GetOutput(), in TEST() 133 EXPECT_THAT(m.GetOutput(), in TEST() 147 EXPECT_THAT(m.GetOutput(), in TEST() 163 EXPECT_THAT(m.GetOutput(), in TEST() 187 m.GetOutput(), in TEST() [all …]
|
D | resize_bilinear_test.cc | 53 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput() function in tflite::__anon96d23e3c0111::ResizeBilinearOpModel 66 EXPECT_THAT(m.GetOutput(), ElementsAreArray(ArrayFloatNear({3, 5, 6}))); in TEST() 71 EXPECT_THAT(const_m.GetOutput(), ElementsAreArray(ArrayFloatNear({3, 5, 6}))); in TEST() 79 EXPECT_THAT(m.GetOutput(), ElementsAreArray(ArrayFloatNear({3, 7, 9}))); in TEST() 84 EXPECT_THAT(const_m.GetOutput(), ElementsAreArray(ArrayFloatNear({3, 7, 9}))); in TEST() 95 EXPECT_THAT(m.GetOutput(), ElementsAreArray(ArrayFloatNear({ in TEST() 107 EXPECT_THAT(const_m.GetOutput(), ElementsAreArray(ArrayFloatNear({ in TEST() 124 EXPECT_THAT(m.GetOutput(), ElementsAreArray(ArrayFloatNear({ in TEST() 141 EXPECT_THAT(const_m.GetOutput(), ElementsAreArray(ArrayFloatNear({ in TEST() 159 EXPECT_THAT(m.GetOutput(), ElementsAreArray(ArrayFloatNear({ in TEST() [all …]
|
D | concatenation_test.cc | 57 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput() function in tflite::__anon4d3a9c4b0111::ConcatenationOpModel 66 std::vector<uint8_t> GetOutput() { return ExtractVector<uint8_t>(output_); } in GetOutput() function in tflite::__anon4d3a9c4b0111::QuantizedConcatenationOpModel 78 EXPECT_THAT(m0.GetOutput(), ElementsAreArray({1, 3, 4, 7})); in TEST() 86 EXPECT_THAT(m0.GetOutput(), ::testing::ElementsAre(5)); in TEST() 94 EXPECT_THAT(m0.GetOutput(), ElementsAreArray({1, 2, 3, 4, 5, 6})); in TEST() 107 EXPECT_THAT(m0.GetOutput(), in TEST() 115 EXPECT_THAT(m0_negative.GetOutput(), in TEST() 123 EXPECT_THAT(m1.GetOutput(), in TEST() 131 EXPECT_THAT(m1_negative.GetOutput(), in TEST() 143 EXPECT_THAT(m0.GetOutput(), in TEST() [all …]
|
D | activations_test.cc | 67 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput() function in tflite::__anon1679b96a0111::FloatActivationsOpModel 83 std::vector<uint8_t> GetOutput() { return ExtractVector<uint8_t>(output_); } in GetOutput() function in tflite::__anon1679b96a0111::QuantizedActivationsOpModel 98 EXPECT_THAT(m.GetOutput(), ElementsAreArray({ in TEST() 112 EXPECT_THAT(m.GetOutput(), ElementsAreArray({ in TEST() 126 EXPECT_THAT(m.GetOutput(), ElementsAreArray({ in TEST() 140 EXPECT_THAT(m.GetOutput(), ElementsAreArray(ArrayFloatNear({ in TEST() 154 EXPECT_THAT(m.GetOutput(), ElementsAreArray(ArrayFloatNear({ in TEST() 176 EXPECT_THAT(m.GetOutput(), in TEST() 188 EXPECT_THAT(m.GetOutput(), ElementsAreArray(ArrayFloatNear({ in TEST() 203 EXPECT_THAT(m2.GetOutput(), ElementsAreArray(ArrayFloatNear({ in TEST() [all …]
|
D | space_to_batch_nd_test.cc | 41 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput() function in tflite::__anon09a4ca450111::SpaceToBatchNDOpModel 107 EXPECT_THAT(m.GetOutput(), ElementsAreArray({1, 3, 9, 11, 2, 4, 10, 12, 5, 7, in TEST() 118 EXPECT_THAT(m.GetOutput(), ElementsAreArray({1, 3, 9, 11, 2, 4, 10, 12, 5, 7, in TEST() 127 EXPECT_THAT(m.GetOutput(), ElementsAreArray({1, 3, 9, 11, 2, 4, 10, 12, 5, 7, in TEST() 138 EXPECT_THAT(m.GetOutput(), ElementsAreArray({1, 3, 9, 11, 2, 4, 10, 12, 5, 7, in TEST() 147 EXPECT_THAT(m.GetOutput(), ElementsAreArray({ in TEST() 160 EXPECT_THAT(m.GetOutput(), ElementsAreArray({ in TEST() 171 EXPECT_THAT(m.GetOutput(), ElementsAreArray({ in TEST() 185 EXPECT_THAT(m.GetOutput(), ElementsAreArray({ in TEST()
|
D | transpose_test.cc | 138 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput() function in tflite::__anon940c88840111::TransposeOpModel 203 EXPECT_THAT(m.GetOutput(), ElementsAreArray({1, 2, 3})); in TEST() 212 EXPECT_THAT(m.GetOutput(), ElementsAreArray({1, 2, 3})); in TEST() 220 EXPECT_THAT(m.GetOutput(), ElementsAreArray({0, 2, 4, 1, 3, 5})); in TEST() 229 EXPECT_THAT(m.GetOutput(), ElementsAreArray({0, 2, 4, 1, 3, 5})); in TEST() 238 EXPECT_THAT(m.GetOutput(), in TEST() 250 EXPECT_THAT(m.GetOutput(), in TEST() 265 EXPECT_THAT(m.GetOutput(), ElementsAreArray({1, 2, 3, 4, 5, 6})); in TEST() 274 EXPECT_THAT(m.GetOutput(), ElementsAreArray({1, 2, 3, 4, 5, 6})); in TEST() 282 EXPECT_THAT(m.GetOutput(), ElementsAreArray({1, 4, 2, 5, 3, 6})); in TEST() [all …]
|
D | pooling_test.cc | 59 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput() function in tflite::__anonc5a88e9d0111::FloatPoolingOpModel 70 std::vector<uint8_t> GetOutput() { return ExtractVector<uint8_t>(output_); } in GetOutput() function in tflite::__anonc5a88e9d0111::QuantizedPoolingOpModel 87 EXPECT_THAT(m.GetOutput(), ElementsAreArray({2.75, 5.75})); in TEST() 106 EXPECT_THAT(m.GetOutput(), ElementsAreArray({44, 92})); in TEST() 119 EXPECT_THAT(m.GetOutput(), ElementsAreArray({6, 10})); in TEST() 138 EXPECT_THAT(m.GetOutput(), ElementsAreArray({96, 160})); in TEST() 151 EXPECT_THAT(m.GetOutput(), ElementsAreArray({3.5, 6.5})); in TEST()
|
D | conv_test.cc | 101 std::vector<float> GetOutput() { return ExtractVector<float>(output_); } in GetOutput() function in tflite::__anon3a8e4ddb0111::ConvolutionOpModel 141 EXPECT_THAT(m.GetOutput(), ElementsAreArray({ in TEST_P() 165 EXPECT_THAT(m.GetOutput(), ElementsAreArray({ in TEST_P() 221 EXPECT_THAT(m.GetOutput(), ElementsAreArray({105, 150, 183, 95, 235, 312, 357, in TEST_P() 275 EXPECT_THAT(m.GetOutput(), ElementsAreArray({115, 160, 193, 105, 245, 322, in TEST_P() 331 EXPECT_THAT(m.GetOutput(), in TEST_P() 374 EXPECT_THAT(m.GetOutput(), ElementsAreArray({312, 357})); in TEST_P() 393 std::vector<uint8_t> GetOutput() { return ExtractVector<uint8_t>(output_); } in GetOutput() function in tflite::__anon3a8e4ddb0111::QuantizedConvolutionOpModel 434 EXPECT_THAT(m.GetOutput(), ElementsAreArray({ in TEST_P() 463 EXPECT_THAT(m.GetOutput(), ElementsAreArray({ in TEST_P()
|
/external/tensorflow/tensorflow/core/kernels/ |
D | quantize_op_test.cc | 51 test::ExpectTensorEqual<quint8>(expected, *GetOutput(0)); in TEST_F() 75 test::ExpectTensorEqual<quint8>(expected, *GetOutput(0)); in TEST_F() 79 test::ExpectTensorEqual<float>(expected_output_min, *GetOutput(1)); in TEST_F() 83 test::ExpectTensorEqual<float>(expected_output_max, *GetOutput(2)); in TEST_F() 105 test::ExpectTensorEqual<quint8>(expected, *GetOutput(0)); in TEST_F() 109 test::ExpectTensorEqual<float>(expected_output_min, *GetOutput(1)); in TEST_F() 113 test::ExpectTensorEqual<float>(expected_output_max, *GetOutput(2)); in TEST_F() 135 test::ExpectTensorEqual<qint8>(expected, *GetOutput(0)); in TEST_F() 139 test::ExpectTensorEqual<float>(expected_output_min, *GetOutput(1)); in TEST_F() 143 test::ExpectTensorEqual<float>(expected_output_max, *GetOutput(2)); in TEST_F() [all …]
|
D | fake_quant_ops_test.cc | 71 Tensor* output = GetOutput(0); in RunTestFakeQuantWithMinMaxArgs() 100 Tensor* output = GetOutput(0); in RunTestFakeQuantWithMinMaxVars() 130 Tensor* output = GetOutput(0); in RunTestFakeQuantWithMinMaxVarsPerChannel() 378 Tensor* output = GetOutput(0); in TEST_F() 408 Tensor* output = GetOutput(0); in TEST_F() 439 Tensor* output = GetOutput(0); in TEST_F() 470 Tensor* output = GetOutput(0); in TEST_F() 667 Tensor* output_bprop_wrt_input = GetOutput(0); in TEST_F() 675 Tensor* output_bprop_wrt_min = GetOutput(1); in TEST_F() 680 Tensor* output_bprop_wrt_max = GetOutput(2); in TEST_F() [all …]
|
D | resize_nearest_neighbor_op_test.cc | 72 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F() 90 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F() 110 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F() 131 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F() 152 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F() 174 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F() 193 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F() 217 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F() 242 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F() 264 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F() [all …]
|
D | dynamic_partition_op_test.cc | 65 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F() 70 test::ExpectTensorEqual<float>(expected, *GetOutput(1)); in TEST_F() 75 test::ExpectTensorEqual<float>(expected, *GetOutput(2)); in TEST_F() 80 test::ExpectTensorEqual<float>(expected, *GetOutput(3)); in TEST_F() 98 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F() 103 test::ExpectTensorEqual<float>(expected, *GetOutput(1)); in TEST_F() 108 test::ExpectTensorEqual<float>(expected, *GetOutput(2)); in TEST_F() 113 test::ExpectTensorEqual<float>(expected, *GetOutput(3)); in TEST_F() 133 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F() 136 test::ExpectTensorEqual<float>(expected_empty, *GetOutput(1)); in TEST_F() [all …]
|
D | parse_tensor_test.cc | 83 ParseSerializedOutput<Eigen::half>(GetOutput(0), &parse_output); in TEST_F() 92 ParseSerializedOutput<float>(GetOutput(0), &parse_output); in TEST_F() 101 ParseSerializedOutput<double>(GetOutput(0), &parse_output); in TEST_F() 110 ParseSerializedOutput<int64>(GetOutput(0), &parse_output); in TEST_F() 119 ParseSerializedOutput<int32>(GetOutput(0), &parse_output); in TEST_F() 128 ParseSerializedOutput<int16>(GetOutput(0), &parse_output); in TEST_F() 137 ParseSerializedOutput<int8>(GetOutput(0), &parse_output); in TEST_F() 146 ParseSerializedOutput<uint16>(GetOutput(0), &parse_output); in TEST_F() 155 ParseSerializedOutput<uint8>(GetOutput(0), &parse_output); in TEST_F() 165 ParseSerializedOutput<complex64>(GetOutput(0), &parse_output); in TEST_F() [all …]
|
D | colorspace_op_test.cc | 48 test::ExpectTensorEqual<T>(expected, *GetOutput(0)); in CheckBlack() 58 test::ExpectTensorEqual<T>(expected, *GetOutput(0)); in CheckGray() 68 test::ExpectTensorEqual<T>(expected, *GetOutput(0)); in CheckWhite() 82 test::ExpectTensorNear<T>(expected, *GetOutput(0), 1e-6); in CheckRedMax() 96 test::ExpectTensorNear<T>(expected, *GetOutput(0), 1e-6); in CheckGreenMax() 110 test::ExpectTensorNear<T>(expected, *GetOutput(0), 1e-6); in CheckBlueMax() 123 test::ExpectTensorNear<T>(expected, *GetOutput(0), 1e-6); in CheckNegativeDifference() 144 test::ExpectTensorEqual<T>(expected, *GetOutput(0)); in CheckBlack() 154 test::ExpectTensorEqual<T>(expected, *GetOutput(0)); in CheckGray() 164 test::ExpectTensorEqual<T>(expected, *GetOutput(0)); in CheckWhite() [all …]
|
D | debug_ops_test.cc | 68 test::ExpectTensorEqual<int32>(expected, *GetOutput(0)); in TEST_F() 93 test::ExpectTensorEqual<int32>(expected, *GetOutput(0)); in TEST_F() 165 test::ExpectTensorEqual<int32>(expected, *GetOutput(0)); in TEST_F() 174 test::ExpectTensorEqual<string>(expected, *GetOutput(0)); in TEST_F() 200 test::ExpectTensorEqual<int64>(expected_nan_count, *GetOutput(0)); in TEST_F() 212 test::ExpectTensorEqual<int64>(expected_nan_count, *GetOutput(0)); in TEST_F() 225 test::ExpectTensorEqual<int64>(expected_nan_count, *GetOutput(0)); in TEST_F() 237 test::ExpectTensorEqual<int64>(expected_nan_count, *GetOutput(0)); in TEST_F() 302 test::ExpectTensorNear<double>(expected, *GetOutput(0), 1e-8); in TEST_F() 341 test::ExpectTensorNear<double>(expected, *GetOutput(0), 1e-8); in TEST_F() [all …]
|
D | non_max_suppression_op_test.cc | 56 test::ExpectTensorEqual<int>(expected, *GetOutput(0)); in TEST_F() 70 test::ExpectTensorEqual<int>(expected, *GetOutput(0)); in TEST_F() 85 test::ExpectTensorEqual<int>(expected, *GetOutput(0)); in TEST_F() 100 test::ExpectTensorEqual<int>(expected, *GetOutput(0)); in TEST_F() 112 test::ExpectTensorEqual<int>(expected, *GetOutput(0)); in TEST_F() 135 test::ExpectTensorEqual<int>(expected, *GetOutput(0)); in TEST_F() 176 test::ExpectTensorEqual<int>(expected, *GetOutput(0)); in TEST_F() 209 test::ExpectTensorEqual<int>(expected, *GetOutput(0)); in TEST_F() 225 test::ExpectTensorEqual<int>(expected, *GetOutput(0)); in TEST_F() 241 test::ExpectTensorEqual<int>(expected, *GetOutput(0)); in TEST_F() [all …]
|
D | control_flow_ops_test.cc | 46 test::ExpectTensorEqual<int32>(expected, *GetOutput(0)); in TEST_F() 47 EXPECT_EQ(nullptr, GetOutput(1)); in TEST_F() 57 test::ExpectTensorEqual<int32>(expected, *GetOutput(1)); in TEST_F() 58 EXPECT_EQ(nullptr, GetOutput(0)); in TEST_F() 68 test::ExpectTensorEqual<int32>(expected, *GetOutput(0)); in TEST_F() 69 EXPECT_EQ(nullptr, GetOutput(1)); in TEST_F() 79 test::ExpectTensorEqual<string>(expected, *GetOutput(1)); in TEST_F() 80 EXPECT_EQ(nullptr, GetOutput(0)); in TEST_F()
|
D | roll_op_test.cc | 63 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F() 78 test::ExpectTensorEqual<string>(expected, *GetOutput(0)); in TEST_F() 99 test::ExpectTensorEqual<std::complex<float>>(expected, *GetOutput(0)); in TEST_F() 116 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F() 134 test::ExpectTensorEqual<string>(expected, *GetOutput(0)); in TEST_F() 150 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F() 168 test::ExpectTensorEqual<string>(expected, *GetOutput(0)); in TEST_F() 185 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F() 203 test::ExpectTensorEqual<string>(expected, *GetOutput(0)); in TEST_F() 219 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F() [all …]
|
D | resize_bilinear_op_test.cc | 124 test::ExpectTensorEqual<float>(*expected, *GetOutput(0)); in TestResize() 179 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F() 189 Tensor* output = GetOutput(0); in TEST_F() 211 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F() 231 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F() 254 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F() 275 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F() 296 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F() 318 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F() 342 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F() [all …]
|
D | fused_batch_norm_op_test.cc | 55 test::ExpectTensorNear<float>(expected, *GetOutput(0), 0.01); in TEST_F() 59 test::ExpectTensorNear<float>(expected_mean, *GetOutput(1), 0.01); in TEST_F() 63 test::ExpectTensorNear<float>(expected_variance, *GetOutput(2), 0.01); in TEST_F() 89 test::ExpectTensorNear<float>(expected, *GetOutput(0), 0.01); in TEST_F() 117 test::ExpectTensorNear<float>(expected_x, *GetOutput(0), 0.01); in TEST_F() 121 test::ExpectTensorNear<float>(expected_scale, *GetOutput(1), 0.01); in TEST_F() 125 test::ExpectTensorNear<float>(expected_offset, *GetOutput(2), 0.01); in TEST_F()
|
D | identity_n_op_test.cc | 47 test::ExpectTensorEqual<int32>(expected0, *GetOutput(0)); in TEST_F() 50 test::ExpectTensorEqual<double>(expected1, *GetOutput(1)); in TEST_F() 60 test::ExpectTensorEqual<int32>(expected, *GetOutput(0)); in TEST_F() 62 test::ExpectTensorEqual<int32>(expected, *GetOutput(1)); in TEST_F() 72 test::ExpectTensorEqual<string>(expected0, *GetOutput(0)); in TEST_F() 75 test::ExpectTensorEqual<int32>(expected1, *GetOutput(1)); in TEST_F()
|
D | crop_and_resize_op_test.cc | 59 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); \ 86 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in REGISTER_TEST() 102 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F() 123 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F() 144 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F() 169 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F() 194 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F() 216 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F() 234 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F() 316 test::ExpectTensorEqual<float>(expected, *GetOutput(0)); in TEST_F()
|
D | quantized_activation_ops_test.cc | 61 const Tensor& output_quantized = *GetOutput(0); in TEST_F() 62 const float output_min = GetOutput(1)->flat<float>()(0); in TEST_F() 63 const float output_max = GetOutput(2)->flat<float>()(0); in TEST_F() 92 const Tensor& output_quantized = *GetOutput(0); in TEST_F() 93 const float output_min = GetOutput(1)->flat<float>()(0); in TEST_F() 94 const float output_max = GetOutput(2)->flat<float>()(0); in TEST_F()
|
D | sparse_add_op_test.cc | 83 test::ExpectTensorEqual<int64>(expected_indices, *GetOutput(0)); in TEST_F() 87 test::ExpectTensorEqual<float>(expected_values, *GetOutput(1)); in TEST_F() 92 test::ExpectTensorEqual<int64>(expected_shape, *GetOutput(2)); in TEST_F() 132 test::ExpectTensorEqual<int64>(expected_indices, *GetOutput(0)); \ 136 test::ExpectTensorEqual<VALTYPE>(expected_values, *GetOutput(1)); \ 141 test::ExpectTensorEqual<int64>(expected_shape, *GetOutput(2)); \ 189 test::ExpectTensorEqual<int64>(expected_indices, *GetOutput(0)); \ 192 test::ExpectTensorEqual<VALTYPE>(expected_values, *GetOutput(1)); \ 197 test::ExpectTensorEqual<int64>(expected_shape, *GetOutput(2)); \
|