Searched refs:argmaxwithvalue_3d (Results 1 – 2 of 2) sorted by relevance
/third_party/mindspore/tests/st/ops/cpu/ |
D | test_argmaxwithvalue_op.py | 77 def argmaxwithvalue_3d(data_type, shape_x): function 124 argmaxwithvalue_3d(np.float32, shape_x) 126 argmaxwithvalue_3d(np.float32, shape_x) 135 argmaxwithvalue_3d(np.float16, shape_x) 144 argmaxwithvalue_3d(np.float32, shape_x) 146 argmaxwithvalue_3d(np.float32, shape_x)
|
/third_party/mindspore/tests/st/ops/gpu/ |
D | test_argmaxwithvalue_op.py | 77 def argmaxwithvalue_3d(data_type, shape_x): function 124 argmaxwithvalue_3d(np.float32, shape_x) 126 argmaxwithvalue_3d(np.float32, shape_x) 135 argmaxwithvalue_3d(np.float16, shape_x) 144 argmaxwithvalue_3d(np.float32, shape_x) 146 argmaxwithvalue_3d(np.float32, shape_x)
|