test_take_along_axis_op_npu.py 4.5 KB