/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
H A D | sub_test.cc | 163 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 D | mul_test.cc | 126 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 D | comparisons_test.cc | 569 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 D | div_test.cc | 108 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 D | squared_difference_test.cc | 96 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 D | add_test.cc | 150 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 D | squared_difference_test.cc | 71 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 D | call_test.cc | 184 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 D | spectral_ops_test.py | 116 def test_shapes(self): member in SpectralOpsTest
|
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
H A D | math_ops_test.cc | 113 auto test_shapes = [&](ShapeInferenceTestOp& op, in TEST() local
|
/aosp_15_r20/external/pytorch/test/ |
H A D | test_sparse.py | 1087 def test_shapes(shapes, dim, fail_message=None): function
|
/aosp_15_r20/external/pytorch/test/torch_np/numpy_tests/core/ |
H A D | test_multiarray.py | 5244 def test_shapes(self): member in MatmulCommon
|