row max (#894)
Showing
- oneflow/core/graph/loss_compute_task_node.cpp 0 additions, 2 deletionsoneflow/core/graph/loss_compute_task_node.cpp
- oneflow/core/kernel/kernel_util.cpp 16 additions, 0 deletionsoneflow/core/kernel/kernel_util.cpp
- oneflow/core/kernel/kernel_util.cu 53 additions, 0 deletionsoneflow/core/kernel/kernel_util.cu
- oneflow/core/kernel/kernel_util.h 32 additions, 0 deletionsoneflow/core/kernel/kernel_util.h
- oneflow/core/kernel/softmax_kernel.cpp 12 additions, 32 deletionsoneflow/core/kernel/softmax_kernel.cpp
- oneflow/core/kernel/softmax_kernel.cu 0 additions, 20 deletionsoneflow/core/kernel/softmax_kernel.cu
- oneflow/core/kernel/softmax_kernel.h 3 additions, 12 deletionsoneflow/core/kernel/softmax_kernel.h
- oneflow/core/kernel/sparse_softmax_cross_entropy_loss_kernel.cpp 1 addition, 11 deletions.../core/kernel/sparse_softmax_cross_entropy_loss_kernel.cpp
- oneflow/core/kernel/sparse_softmax_cross_entropy_loss_kernel.h 0 additions, 2 deletions...ow/core/kernel/sparse_softmax_cross_entropy_loss_kernel.h
- oneflow/core/operator/loss_op.cpp 2 additions, 2 deletionsoneflow/core/operator/loss_op.cpp
- oneflow/core/operator/loss_op.h 7 additions, 2 deletionsoneflow/core/operator/loss_op.h
- oneflow/core/operator/operator.cpp 4 additions, 1 deletiononeflow/core/operator/operator.cpp
- oneflow/core/operator/operator.h 2 additions, 0 deletionsoneflow/core/operator/operator.h
- oneflow/core/operator/softmax_op.cpp 2 additions, 6 deletionsoneflow/core/operator/softmax_op.cpp
- oneflow/core/operator/softmax_op.h 1 addition, 1 deletiononeflow/core/operator/softmax_op.h
- oneflow/core/operator/sparse_softmax_cross_entropy_loss_op.cpp 3 additions, 9 deletions...ow/core/operator/sparse_softmax_cross_entropy_loss_op.cpp
- oneflow/core/operator/sparse_softmax_cross_entropy_loss_op.h 1 addition, 1 deletiononeflow/core/operator/sparse_softmax_cross_entropy_loss_op.h
Please register or sign in to comment