arg_max_op.cu 1.0 KB