Home
last modified time | relevance | path

Searched defs:test_shapes (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dsub_test.cc163 std::vector<std::vector<int>> test_shapes = { in TEST() local
180 std::vector<std::vector<int>> test_shapes = { in TEST() local
197 std::vector<std::vector<int>> test_shapes = {{1, 3, 1, 2, 1}}; in TEST() local
233 std::vector<std::vector<int>> test_shapes = { in TEST() local
248 std::vector<std::vector<int>> test_shapes = { in TEST() local
284 std::vector<std::vector<int>> test_shapes = { in TEST() local
299 std::vector<std::vector<int>> test_shapes = { in TEST() local
389 std::vector<std::vector<int>> test_shapes = { in QuantizedVariousInputShapes() local
446 std::vector<std::vector<int>> test_shapes = { in QuantizedWithBroadcast() local
523 std::vector<std::vector<int>> test_shapes = { in TEST() local
[all …]
H A Dmul_test.cc126 const std::vector<std::vector<int>> test_shapes = { in TEST() local
143 const std::vector<std::vector<int>> test_shapes = { in TEST() local
160 const std::vector<std::vector<int>> test_shapes = { in TEST() local
179 const std::vector<std::vector<int>> test_shapes = { in TEST() local
227 const std::vector<std::vector<int>> test_shapes = { in TEST() local
273 const std::vector<std::vector<int>> test_shapes = { in TEST() local
308 const std::vector<std::vector<int>> test_shapes = { in TEST() local
323 const std::vector<std::vector<int>> test_shapes = { in TEST() local
451 const std::vector<std::vector<int>> test_shapes = { in WithBroadcast() local
480 const std::vector<std::vector<int>> test_shapes = { in QuantizedWithMixedBroadcast() local
H A Dcomparisons_test.cc569 std::vector<std::vector<int>> test_shapes = { in TEST() local
587 std::vector<std::vector<int>> test_shapes = { in TEST() local
605 std::vector<std::vector<int>> test_shapes = { in TEST() local
623 std::vector<std::vector<int>> test_shapes = { in TEST() local
641 std::vector<std::vector<int>> test_shapes = { in TEST() local
660 std::vector<std::vector<int>> test_shapes = { in TEST() local
679 std::vector<std::vector<int>> test_shapes = { in TEST() local
697 std::vector<std::vector<int>> test_shapes = { in TEST() local
715 std::vector<std::vector<int>> test_shapes = { in TEST() local
733 std::vector<std::vector<int>> test_shapes = { in TEST() local
[all …]
H A Ddiv_test.cc108 std::vector<std::vector<int>> test_shapes = { in TEST() local
125 std::vector<std::vector<int>> test_shapes = { in TEST() local
143 std::vector<std::vector<int>> test_shapes = {{1, 2, 1, 2, 2}}; in TEST() local
180 std::vector<std::vector<int>> test_shapes = { in TEST() local
195 std::vector<std::vector<int>> test_shapes = { in TEST() local
260 const std::vector<std::vector<int>> test_shapes = { in QuantizedVariousInputShapes() local
287 const std::vector<std::vector<int>> test_shapes = { in QuantizedWithBroadcast() local
H A Dsquared_difference_test.cc96 std::vector<std::vector<int>> test_shapes = { in TEST() local
113 std::vector<std::vector<int>> test_shapes = { in TEST() local
141 std::vector<std::vector<int>> test_shapes = { in TEST() local
156 std::vector<std::vector<int>> test_shapes = { in TEST() local
187 std::vector<std::vector<int>> test_shapes = { in TEST() local
205 std::vector<std::vector<int>> test_shapes = { in TEST() local
H A Dadd_test.cc150 std::vector<std::vector<int>> test_shapes = { in TEST() local
166 std::vector<std::vector<int>> test_shapes = { in TEST() local
198 std::vector<std::vector<int>> test_shapes = { in TEST() local
278 std::vector<std::vector<int>> test_shapes = { in TYPED_TEST() local
293 std::vector<std::vector<int>> test_shapes = { in TYPED_TEST() local
410 std::vector<std::vector<int>> test_shapes = { in QuantizedVariousInputShapes() local
440 std::vector<std::vector<int>> test_shapes = { in QuantizedWithScalarBroadcast() local
491 std::vector<std::vector<int>> test_shapes = { in QuantizedWithMixedBroadcast() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
H A Dsquared_difference_test.cc71 std::vector<std::vector<int>> test_shapes = { in TEST() local
88 std::vector<std::vector<int>> test_shapes = { in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/
H A Dcall_test.cc184 std::vector<std::vector<int>> test_shapes = { in TEST_F() local
258 std::vector<std::vector<int>> test_shapes = { in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/signal/
H A Dspectral_ops_test.py116 def test_shapes(self): member in SpectralOpsTest
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dmath_ops_test.cc113 auto test_shapes = [&](ShapeInferenceTestOp& op, in TEST() local
/aosp_15_r20/external/pytorch/test/
H A Dtest_sparse.py1087 def test_shapes(shapes, dim, fail_message=None): function
/aosp_15_r20/external/pytorch/test/torch_np/numpy_tests/core/
H A Dtest_multiarray.py5244 def test_shapes(self): member in MatmulCommon