Searched refs:test_squeeze (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/tests/st/ops/graph_kernel/ |
| D | test_squeeze.py | 39 def test_squeeze(shape, dtype, axis=()): function
51 test_squeeze((1, 16, 1, 1), np.int32)
52 test_squeeze((1, 16, 1, 1), np.float32, (0, 2))
|
| /third_party/mindspore/tests/ut/python/ops/ |
| D | test_array_ops.py | 104 def test_squeeze(): function
|
| /third_party/mindspore/tests/st/numpy_native/ |
| D | test_array_ops.py | 203 def test_squeeze(): function
|