/external/tensorflow/tensorflow/lite/kernels/ |
fake_quant_test.cc | 24 using ::testing::ElementsAreArray; 60 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 1, 2})); 63 ElementsAreArray(ArrayFloatNear({0, 1, 0.25098, 0.498039, 0.443137, 0}))); 73 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 1, 2})); 75 ElementsAreArray(ArrayFloatNear( 86 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 1, 2})); 88 ElementsAreArray(ArrayFloatNear( 99 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 1, 2})); 101 ElementsAreArray(ArrayFloatNear(
|
ceil_test.cc | 25 using ::testing::ElementsAreArray; 52 EXPECT_THAT(model.GetOutput(), ElementsAreArray({9, 0})); 53 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({2})); 72 ElementsAreArray({1, 9, 1, 10, 1, 0, -8, 0, -9, 0})); 73 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({2, 1, 1, 5}));
|
floor_test.cc | 25 using ::testing::ElementsAreArray; 52 EXPECT_THAT(model.GetOutput(), ElementsAreArray({8, 0})); 53 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({2})); 72 ElementsAreArray({0, 8, 0, 9, 0, -1, -9, -1, -10, -1})); 73 EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({2, 1, 1, 5}));
|
local_response_norm_test.cc | 24 using ::testing::ElementsAreArray; 59 ElementsAreArray(ArrayFloatNear({-0.55, 0.3, 0.35, 0.6, -0.35, 0.05}))); 68 EXPECT_THAT(m.GetOutput(), ElementsAreArray(ArrayFloatNear( 80 ElementsAreArray(ArrayFloatNear({-0.22, 0.12, 0.14, 0.24, -0.14, 0.02}))); 90 ElementsAreArray(ArrayFloatNear(
|
rank_test.cc | 27 using ::testing::ElementsAreArray; 56 EXPECT_THAT(model.GetOutput(), ElementsAreArray({5})); 64 EXPECT_THAT(model.GetOutput(), ElementsAreArray({5})); 72 EXPECT_THAT(model.GetOutput(), ElementsAreArray({0})); 80 EXPECT_THAT(model.GetOutput(), ElementsAreArray({2}));
|
fill_test.cc | 24 using ::testing::ElementsAreArray; 54 ElementsAreArray({-11, -11, -11, -11, -11, -11})); 55 EXPECT_THAT(m.GetTensorShape(m.output()), ElementsAreArray({2, 3})); 64 ElementsAreArray({2 ^ 45, 2 ^ 45, 2 ^ 45, 2 ^ 45, 2 ^ 45, 2 ^ 45, 66 EXPECT_THAT(m.GetTensorShape(m.output()), ElementsAreArray({2, 4})); 75 ElementsAreArray({4.0, 4.0, 4.0, 4.0, 4.0, 4.0, 4.0, 4.0})); 76 EXPECT_THAT(m.GetTensorShape(m.output()), ElementsAreArray({2, 2, 2})); 83 EXPECT_THAT(m.ExtractVector<float>(m.output()), ElementsAreArray({4.0}));
|
where_test.cc | 26 using ::testing::ElementsAreArray; 63 EXPECT_THAT(m.GetOutput(), ElementsAreArray({0, 2})); 79 EXPECT_THAT(m.GetOutput(), ElementsAreArray({0, 0, // 89 EXPECT_THAT(m.GetOutput(), ElementsAreArray({0, 0, // 102 EXPECT_THAT(m.GetOutput(), ElementsAreArray({0, 0, // 124 EXPECT_THAT(m.GetOutput(), ElementsAreArray({0, 0, 0, // 134 EXPECT_THAT(m.GetOutput(), ElementsAreArray({0, 0, 0, // 146 EXPECT_THAT(m.GetOutput(), ElementsAreArray({0, 0, 0, //
|
gather_nd_test.cc | 26 using ::testing::ElementsAreArray; 68 EXPECT_THAT(m.GetOutput<float>(), ElementsAreArray({1.1, 2.2})); 77 EXPECT_THAT(m.GetOutput<float>(), ElementsAreArray({2.1, 2.2, 1.1, 1.2})); 87 EXPECT_THAT(m.GetOutput<float>(), ElementsAreArray({2.1, 2.2, 1.1, 1.2})); 97 EXPECT_THAT(m.GetOutput<float>(), ElementsAreArray({1.1, 2.2})); 106 EXPECT_THAT(m.GetOutput<float>(), ElementsAreArray({1.1, 1.1})); 118 EXPECT_THAT(m.GetOutput<float>(), ElementsAreArray({-1.2, -4.1})); 131 ElementsAreArray({1.1, -1.2, 1.3, -2.1, 2.2, 2.3, 5.1, -5.2, 5.3, 144 EXPECT_THAT(m.GetOutput<float>(), ElementsAreArray({-1.2, -4.1})); 157 ElementsAreArray({3.1, 3.2, -3.3, -4.1, -4.2, 4.3, 1.1, -1.2, 1.3 [all...] |
transpose_test.cc | 26 using ::testing::ElementsAreArray; 227 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3})); 228 EXPECT_THAT(m.GetOutput(), ElementsAreArray({1, 2, 3})); 236 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3})); 237 EXPECT_THAT(m.GetOutput(), ElementsAreArray({1, 2, 3})); 244 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 3})); 245 EXPECT_THAT(m.GetOutput(), ElementsAreArray({0, 2, 4, 1, 3, 5})); 253 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({2, 3})); 254 EXPECT_THAT(m.GetOutput(), ElementsAreArray({0, 2, 4, 1, 3, 5})); 262 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({4, 2, 3})) [all...] |
pack_test.cc | 25 using ::testing::ElementsAreArray; 62 EXPECT_THAT(model.GetOutput(), ElementsAreArray({1, 4, 2, 5, 3, 6})); 72 EXPECT_THAT(model.GetOutput(), ElementsAreArray({1, 2, 3, 4, 5, 6})); 82 ElementsAreArray({1, 2, 3, 7, 8, 9, 4, 5, 6, 10, 11, 12})); 93 ElementsAreArray({1, 2, 3, 4, 5, 6, 7, 8, 17, 18, 19, 106 EXPECT_THAT(model.GetOutput(), ElementsAreArray({1, 4, 2, 5, 3, 6})); 116 EXPECT_THAT(model.GetOutput(), ElementsAreArray({1, 2, 3, 4, 5, 6})); 126 ElementsAreArray({1, 2, 3, 7, 8, 9, 4, 5, 6, 10, 11, 12})); 138 ElementsAreArray({1LL << 33, 4LL, 2LL, 5LL, 3LL, -(1LL << 34)})); 149 ElementsAreArray({1LL << 33, 2LL, 3LL, 4LL, 5LL, -(1LL << 34)})) [all...] |
neg_test.cc | 24 using ::testing::ElementsAreArray; 56 ElementsAreArray({2.0f, 1.0f, 0.f, -1.0f, -2.0f, -3.0f})); 63 EXPECT_THAT(m.GetOutput<int32_t>(), ElementsAreArray({2, 1, 0, -1, -2, -3})); 70 EXPECT_THAT(m.GetOutput<int64_t>(), ElementsAreArray({2, 1, 0, -1, -2, -3}));
|
mirror_pad_test.cc | 24 using ::testing::ElementsAreArray; 59 EXPECT_THAT(model.GetOutput(), ElementsAreArray({1, 2, 3, 4, 5, 6})); 70 ElementsAreArray({1, 2, 3, 2, 4, 5, 6, 5, 1, 2, 3, 2})); 81 ElementsAreArray({5, 4, 5, 6, 2, 1, 2, 3, 5, 4, 5, 6})); 92 ElementsAreArray({1, 2, 3, 3, 4, 5, 6, 6, 4, 5, 6, 6})); 103 ElementsAreArray({1, 1, 2, 3, 1, 1, 2, 3, 4, 4, 5, 6})); 114 ElementsAreArray({1, 1, 2, 3, 3, 1, 1, 2, 3, 3, 126 ElementsAreArray({5, 4, 5, 6, 5, 2, 1, 2, 3, 2, 139 ElementsAreArray({6, 5, 4, 4, 5, 6, 6, 5, 4, 3, 2, 1, 1, 2, 3, 3, 2, 1, 152 ElementsAreArray({6, 5, 4, 5, 6, 5, 4, 3, 2, 1, 2, 3, 2, 1 [all...] |
pooling_test.cc | 25 using ::testing::ElementsAreArray; 128 EXPECT_THAT(m.GetOutput(), ElementsAreArray({2.75, 5.75})); 146 ElementsAreArray(ArrayFloatNear({2.75, 5.75}))); 147 EXPECT_THAT(m.GetOutput(), ElementsAreArray({44, 92})); 165 EXPECT_THAT(m.GetDequantizedOutput(), ElementsAreArray(ArrayFloatNear({16}))); 188 ElementsAreArray(ArrayFloatNear( 212 ElementsAreArray(ArrayFloatNear({2.75, 5.75}))); 213 EXPECT_THAT(m.GetOutput(), ElementsAreArray({44 - 128, 92 - 128})); 247 EXPECT_THAT(m.GetOutput(), ElementsAreArray({6, 10})); 265 ElementsAreArray(ArrayFloatNear({6, 10}))) [all...] |
batch_to_space_nd_test.cc | 25 using ::testing::ElementsAreArray; 107 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 4, 4, 1})); 109 ElementsAreArray( 118 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 4, 4, 1})); 120 ElementsAreArray( 130 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 4, 4, 1})); 132 ElementsAreArray( 142 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({1, 4, 4, 1})); 144 ElementsAreArray(
|
detection_postprocess_test.cc | 35 using ::testing::ElementsAreArray; 169 ElementsAreArray(ArrayFloatNear( 176 ElementsAreArray(ArrayFloatNear({1, 0, 0}, 1e-1))); 181 ElementsAreArray(ArrayFloatNear({0.95, 0.9, 0.3}, 1e-1))); 186 ElementsAreArray(ArrayFloatNear({3.0}, 1e-1))); 228 ElementsAreArray(ArrayFloatNear( 235 ElementsAreArray(ArrayFloatNear({1, 0, 0}, 1e-1))); 240 ElementsAreArray(ArrayFloatNear({0.95, 0.9, 0.3}, 1e-1))); 245 ElementsAreArray(ArrayFloatNear({3.0}, 1e-1))); 380 ElementsAreArray(ArrayFloatNear [all...] |
div_test.cc | 24 using ::testing::ElementsAreArray; 70 ElementsAreArray(ArrayFloatNear({-0.4, 1.0, 0.8, 1.6}))); 81 ElementsAreArray(ArrayFloatNear({-1.0, 1.0, 0.8, 1.0}))); 96 ElementsAreArray(ArrayFloatNear({-20.0, 1.0, 0.5, 1.6, -1.0, 20.0}))) 113 ElementsAreArray(ArrayFloatNear({-2.0, 2.0, 0.7, 0.8, 1.1, -1.23}))) 125 EXPECT_THAT(m.GetOutput(), ElementsAreArray({0, -1, 5, 1})); 135 EXPECT_THAT(m.GetOutput(), ElementsAreArray({-1, 1, 0, 1})); 148 EXPECT_THAT(m.GetOutput(), ElementsAreArray({-20, 1, 0, 1, -1, 20})) 163 EXPECT_THAT(m.GetOutput(), ElementsAreArray({-6, 7, 2, 2, 3, -41}))
|
space_to_depth_test.cc | 25 using ::testing::ElementsAreArray; 65 ElementsAreArray({1.4, 2.3, 3.2, 4.1, 5.4, 6.3, 7.2, 8.1})); 73 EXPECT_THAT(m.GetOutput<uint8_t>(), ElementsAreArray({1, 2, 3, 4})); 81 EXPECT_THAT(m.GetOutput<int8_t>(), ElementsAreArray({1, 2, 3, 4})); 90 ElementsAreArray({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12})); 99 ElementsAreArray(
|
embedding_lookup_test.cc | 33 using ::testing::ElementsAreArray; 105 ElementsAreArray(ArrayFloatNear({ 124 ElementsAreArray(ArrayFloatNear( 145 ElementsAreArray(ArrayFloatNear( 166 ElementsAreArray(ArrayFloatNear( 187 ElementsAreArray(ArrayFloatNear( 208 ElementsAreArray(ArrayFloatNear( 229 ElementsAreArray(ArrayFloatNear(
|
concatenation_test.cc | 25 using ::testing::ElementsAreArray; 101 EXPECT_THAT(m0.GetOutput(), ElementsAreArray({1, 3, 4, 7})); 117 EXPECT_THAT(m0.GetOutput(), ElementsAreArray({1, 2, 3, 4, 5, 6})); 131 ElementsAreArray({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12})); 139 ElementsAreArray({1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12})); 147 ElementsAreArray({1, 2, 3, 7, 8, 9, 4, 5, 6, 10, 11, 12})); 155 ElementsAreArray({1, 2, 3, 7, 8, 9, 4, 5, 6, 10, 11, 12})); 167 ElementsAreArray({ 184 ElementsAreArray(ArrayFloatNear({ 189 ElementsAreArray({ [all...] |
dequantize_test.cc | 27 using ::testing::ElementsAreArray; 61 ElementsAreArray(ArrayFloatNear( 72 ElementsAreArray(ArrayFloatNear(
|
exp_test.cc | 24 using ::testing::ElementsAreArray; 57 EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({3, 1, 2})); 59 ElementsAreArray(ArrayFloatNear(
|
/external/tensorflow/tensorflow/lite/models/smartreply/ops/ |
normalize_test.cc | 33 using ::testing::ElementsAreArray; 66 ElementsAreArray({"<S> i am good; you are welcome <E>"})); 72 EXPECT_THAT(m.GetStringOutput(), ElementsAreArray({"<S> hi ! <E>"})); 78 EXPECT_THAT(m.GetStringOutput(), ElementsAreArray({"<S> <E>"}));
|
/external/libtextclassifier/utils/ |
token-feature-extractor_test.cc | 53 testing::ElementsAreArray({ 73 EXPECT_THAT(dense_features, testing::ElementsAreArray({1.0, 1.0})); 81 testing::ElementsAreArray({ 104 EXPECT_THAT(dense_features, testing::ElementsAreArray({-1.0, 0.0})); 123 testing::ElementsAreArray({extractor.HashToken("^Hello$")})); 124 EXPECT_THAT(dense_features, testing::ElementsAreArray({1.0, 1.0})); 132 testing::ElementsAreArray({extractor.HashToken("^world!$")})); 133 EXPECT_THAT(dense_features, testing::ElementsAreArray({-1.0, 0.0})); 152 testing::ElementsAreArray({ 172 EXPECT_THAT(dense_features, testing::ElementsAreArray({1.0, 1.0})) [all...] |
/external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/ |
lstm_utils_test.cc | 120 EXPECT_THAT(GetDstBuffer()->data, ElementsAreArray(ArrayFloatNear(expected))); 130 EXPECT_THAT(GetDstBuffer()->data, ElementsAreArray(ArrayFloatNear(expected))); 140 EXPECT_THAT(GetDstBuffer()->data, ElementsAreArray(ArrayFloatNear(expected))); 150 EXPECT_THAT(GetDstBuffer()->data, ElementsAreArray(ArrayFloatNear(expected))); 168 EXPECT_THAT(GetDstBuffer()->data, ElementsAreArray(ArrayFloatNear(expected))); 179 EXPECT_THAT(GetDstBuffer()->data, ElementsAreArray(ArrayFloatNear(expected))); 190 EXPECT_THAT(GetDstBuffer()->data, ElementsAreArray(ArrayFloatNear(expected))); 201 EXPECT_THAT(GetDstBuffer()->data, ElementsAreArray(ArrayFloatNear(expected))); 224 EXPECT_THAT(GetDstBuffer()->data, ElementsAreArray(ArrayFloatNear(expected))); 233 EXPECT_THAT(GetDstBuffer()->data, ElementsAreArray(ArrayFloatNear(expected))) [all...] |
/external/tensorflow/tensorflow/lite/tools/ |
gen_op_registration_test.cc | 20 using ::testing::ElementsAreArray; 47 EXPECT_THAT(builtin_ops_, ElementsAreArray({"CONV_2D"})); 48 EXPECT_THAT(custom_ops_, ElementsAreArray({"testing_op"}));
|