arg_max_op.cu 1.3 KB