Home
last modified time | relevance | path

Searched refs:argmaxwithvalue_3d (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/tests/st/ops/cpu/
Dtest_argmaxwithvalue_op.py77 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/
Dtest_argmaxwithvalue_op.py77 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)