Home
last modified time | relevance | path

Searched refs:test_shapes (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dsquared_difference_test.cc95 std::vector<std::vector<int>> test_shapes = { in TEST() local
97 for (int i = 0; i < test_shapes.size(); ++i) { in TEST()
98 FloatSquaredDifferenceOpModel m({TensorType_FLOAT32, test_shapes[i]}, in TEST()
99 {TensorType_FLOAT32, test_shapes[i]}, in TEST()
112 std::vector<std::vector<int>> test_shapes = { in TEST() local
114 for (int i = 0; i < test_shapes.size(); ++i) { in TEST()
116 {TensorType_FLOAT32, test_shapes[i]}, in TEST()
140 std::vector<std::vector<int>> test_shapes = { in TEST() local
142 for (int i = 0; i < test_shapes.size(); ++i) { in TEST()
143 IntegerSquaredDifferenceOpModel m({TensorType_INT32, test_shapes[i]}, in TEST()
[all …]
Ddiv_test.cc108 std::vector<std::vector<int>> test_shapes = { in TEST() local
110 for (int i = 0; i < test_shapes.size(); ++i) { in TEST()
111 FloatDivOpModel m({TensorType_FLOAT32, test_shapes[i]}, in TEST()
112 {TensorType_FLOAT32, test_shapes[i]}, in TEST()
125 std::vector<std::vector<int>> test_shapes = { in TEST() local
127 for (int i = 0; i < test_shapes.size(); ++i) { in TEST()
128 FloatDivOpModel m({TensorType_FLOAT32, test_shapes[i]}, in TEST()
143 std::vector<std::vector<int>> test_shapes = {{1, 2, 1, 2, 2}}; in TEST() local
144 for (int i = 0; i < test_shapes.size(); ++i) { in TEST()
145 FloatDivOpModel m({TensorType_FLOAT32, test_shapes[i]}, in TEST()
[all …]
Dmul_test.cc116 const std::vector<std::vector<int>> test_shapes = { in TEST() local
118 for (int i = 0; i < test_shapes.size(); ++i) { in TEST()
119 FloatMulOpModel m({TensorType_FLOAT32, test_shapes[i]}, in TEST()
120 {TensorType_FLOAT32, test_shapes[i]}, in TEST()
133 const std::vector<std::vector<int>> test_shapes = { in TEST() local
135 for (int i = 0; i < test_shapes.size(); ++i) { in TEST()
136 FloatMulOpModel m({TensorType_FLOAT32, test_shapes[i]}, in TEST()
150 const std::vector<std::vector<int>> test_shapes = { in TEST() local
152 for (int i = 0; i < test_shapes.size(); ++i) { in TEST()
153 FloatMulOpModel m({TensorType_FLOAT32, test_shapes[i]}, in TEST()
[all …]
Dsub_test.cc138 std::vector<std::vector<int>> test_shapes = { in TEST() local
140 for (int i = 0; i < test_shapes.size(); ++i) { in TEST()
141 FloatSubOpModel m({TensorType_FLOAT32, test_shapes[i]}, in TEST()
142 {TensorType_FLOAT32, test_shapes[i]}, in TEST()
155 std::vector<std::vector<int>> test_shapes = { in TEST() local
157 for (int i = 0; i < test_shapes.size(); ++i) { in TEST()
158 FloatSubOpModel m({TensorType_FLOAT32, test_shapes[i]}, in TEST()
172 std::vector<std::vector<int>> test_shapes = {{1, 3, 1, 2, 1}}; in TEST() local
173 for (int i = 0; i < test_shapes.size(); ++i) { in TEST()
174 FloatSubOpModel m({TensorType_FLOAT32, test_shapes[i]}, in TEST()
[all …]
Dadd_test.cc135 std::vector<std::vector<int>> test_shapes = { in TEST() local
137 for (size_t i = 0; i < test_shapes.size(); ++i) { in TEST()
138 FloatAddOpModel m({TensorType_FLOAT32, test_shapes[i]}, in TEST()
139 {TensorType_FLOAT32, test_shapes[i]}, in TEST()
151 std::vector<std::vector<int>> test_shapes = { in TEST() local
153 for (size_t i = 0; i < test_shapes.size(); ++i) { in TEST()
154 FloatAddOpModel m({TensorType_FLOAT32, test_shapes[i]}, in TEST()
183 std::vector<std::vector<int>> test_shapes = { in TEST() local
198 for (size_t i = 0; i < test_shapes.size(); ++i) { in TEST()
200 {TensorType_FLOAT32, base_shape}, {TensorType_FLOAT32, test_shapes[i]}, in TEST()
[all …]
Dcomparisons_test.cc569 std::vector<std::vector<int>> test_shapes = { in TEST() local
571 for (int i = 0; i < test_shapes.size(); ++i) { in TEST()
572 ComparisonOpModel model({TensorType_UINT8, test_shapes[i], kMin, kMax}, in TEST()
587 std::vector<std::vector<int>> test_shapes = { in TEST() local
589 for (int i = 0; i < test_shapes.size(); ++i) { in TEST()
590 ComparisonOpModel model({TensorType_UINT8, test_shapes[i], kMin, kMax}, in TEST()
605 std::vector<std::vector<int>> test_shapes = { in TEST() local
607 for (int i = 0; i < test_shapes.size(); ++i) { in TEST()
608 ComparisonOpModel model({TensorType_INT8, test_shapes[i], kMin, kMax}, in TEST()
623 std::vector<std::vector<int>> test_shapes = { in TEST() local
[all …]
/external/tensorflow/tensorflow/lite/micro/kernels/
Ddiv_test.cc106 std::vector<std::vector<int>> test_shapes = { in TEST() local
108 for (int i = 0; i < test_shapes.size(); ++i) { in TEST()
109 FloatDivOpModel m({TensorType_FLOAT32, test_shapes[i]}, in TEST()
110 {TensorType_FLOAT32, test_shapes[i]}, in TEST()
123 std::vector<std::vector<int>> test_shapes = { in TEST() local
125 for (int i = 0; i < test_shapes.size(); ++i) { in TEST()
126 FloatDivOpModel m({TensorType_FLOAT32, test_shapes[i]}, in TEST()
141 std::vector<std::vector<int>> test_shapes = {{1, 2, 1, 2, 2}}; in TEST() local
142 for (int i = 0; i < test_shapes.size(); ++i) { in TEST()
143 FloatDivOpModel m({TensorType_FLOAT32, test_shapes[i]}, in TEST()
[all …]
Dadd_test.cc186 const int test_shapes[num_shapes][max_shape_size] = { in TF_LITE_MICRO_TEST() local
194 tflite::testing::TestAddFloat(test_shapes[i], input1_values, test_shapes[i], in TF_LITE_MICRO_TEST()
195 input2_values, test_shapes[i], in TF_LITE_MICRO_TEST()
211 const int test_shapes[num_shapes][max_shape_size] = { in TF_LITE_MICRO_TEST() local
219 tflite::testing::TestAddFloat(test_shapes[i], input1_values, input2_shape, in TF_LITE_MICRO_TEST()
220 input2_values, test_shapes[i], in TF_LITE_MICRO_TEST()
316 const int test_shapes[num_shapes][max_shape_size] = { in TF_LITE_MICRO_TEST() local
334 test_shapes[i], input1_values, input1_quantized, scales[0], in TF_LITE_MICRO_TEST()
335 zero_points[0], test_shapes[i], input2_values, input2_quantized, in TF_LITE_MICRO_TEST()
336 scales[1], zero_points[1], test_shapes[i], golden_values, in TF_LITE_MICRO_TEST()
[all …]
Dsub_test.cc185 const int test_shapes[num_shapes][max_shape_size] = { in TF_LITE_MICRO_TEST() local
193 tflite::testing::TestSubFloat(test_shapes[i], input1_values, test_shapes[i], in TF_LITE_MICRO_TEST()
194 input2_values, test_shapes[i], in TF_LITE_MICRO_TEST()
210 const int test_shapes[num_shapes][max_shape_size] = { in TF_LITE_MICRO_TEST() local
218 tflite::testing::TestSubFloat(test_shapes[i], input1_values, input2_shape, in TF_LITE_MICRO_TEST()
219 input2_values, test_shapes[i], in TF_LITE_MICRO_TEST()
315 const int test_shapes[num_shapes][max_shape_size] = { in TF_LITE_MICRO_TEST() local
333 test_shapes[i], input1_values, input1_quantized, scales[0], in TF_LITE_MICRO_TEST()
334 zero_points[0], test_shapes[i], input2_values, input2_quantized, in TF_LITE_MICRO_TEST()
335 scales[1], zero_points[1], test_shapes[i], golden_values, in TF_LITE_MICRO_TEST()
[all …]
Dcomparisons_test.cc793 int test_shapes[num_shapes][max_shape_size] = { in TF_LITE_MICRO_TEST() local
797 int* input1_dim = test_shapes[i]; in TF_LITE_MICRO_TEST()
822 int test_shapes[num_shapes][max_shape_size] = { in TF_LITE_MICRO_TEST() local
826 int* input1_dim = test_shapes[i]; in TF_LITE_MICRO_TEST()
851 int test_shapes[num_shapes][max_shape_size] = { in TF_LITE_MICRO_TEST() local
855 int* input1_dim = test_shapes[i]; in TF_LITE_MICRO_TEST()
880 int test_shapes[num_shapes][max_shape_size] = { in TF_LITE_MICRO_TEST() local
884 int* input1_dim = test_shapes[i]; in TF_LITE_MICRO_TEST()
909 int test_shapes[num_shapes][max_shape_size] = { in TF_LITE_MICRO_TEST() local
913 int* input1_dim = test_shapes[i]; in TF_LITE_MICRO_TEST()
[all …]
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Dsquared_difference_test.cc71 std::vector<std::vector<int>> test_shapes = { in TEST() local
73 for (int i = 0; i < test_shapes.size(); ++i) { in TEST()
74 SquaredDifferenceOpModel m({TensorType_INT8, test_shapes[i], -2.0, 1.7}, in TEST()
75 {TensorType_INT8, test_shapes[i], -1.0, 1.0}, in TEST()
88 std::vector<std::vector<int>> test_shapes = { in TEST() local
91 for (int i = 0; i < test_shapes.size(); ++i) { in TEST()
92 SquaredDifferenceOpModel m({TensorType_INT8, test_shapes[i], -0.2, 1.1}, in TEST()
/external/tensorflow/tensorflow/core/ops/
Dmath_ops_test.cc113 auto test_shapes = [&](ShapeInferenceTestOp& op, in TEST() local
177 test_shapes(op, true); in TEST()
182 test_shapes(op, false); in TEST()
/external/tensorflow/tensorflow/python/kernel_tests/
Dinplace_ops_test.py172 test_shapes = [(), (1,), (2, 3), (0, 2), (2, 3, 5), (2, 0, 5)]
173 for shape in test_shapes:
/external/tensorflow/tensorflow/python/kernel_tests/signal/
Dspectral_ops_test.py120 def test_shapes(self): member in SpectralOpsTest
/external/tensorflow/tensorflow/python/ops/numpy_ops/
Dnp_array_ops_test.py1108 test_shapes = [(), (1,), (2, 3, 4), (2, 3, 0, 4)]
1111 for shape in test_shapes: