registry_callback_fn return maybe (#5456)
* modified SetInputArgModifyFn
* Delete the CHECK changes in the assign_op.cpp file
* Format
* Modified the OutputArgModifyFn interface
* add return
* maybe error stack from CheckAndConstructOp to OutputArgModifier callback function
* maybe error stack from CheckAndConstructOp to OutputArgModifier callback function
* OutputArgModifier return maybe part_1
* maybe error stack from CheckAndConstructOp to OutputArgModifier callback function
* input_arg_modifier return maybe
* gen_bw_fn return maybe
* bw_gen_fn return maybe
* registry_callback_fn return maybe
* fix bug after merge master
* fix bug
Co-authored-by:
aishangjj <702572275@qq.com>
Showing
- oneflow/user/ops/same_padding_op.cpp 29 additions, 26 deletionsoneflow/user/ops/same_padding_op.cpp
- oneflow/user/ops/sigmoid_op.cpp 1 addition, 1 deletiononeflow/user/ops/sigmoid_op.cpp
- oneflow/user/ops/softmax_op.cpp 1 addition, 1 deletiononeflow/user/ops/softmax_op.cpp
- oneflow/user/ops/squeeze_op.cpp 2 additions, 2 deletionsoneflow/user/ops/squeeze_op.cpp
- oneflow/user/ops/test_ops.cpp 4 additions, 4 deletionsoneflow/user/ops/test_ops.cpp
- oneflow/user/ops/transpose_ops.cpp 3 additions, 3 deletionsoneflow/user/ops/transpose_ops.cpp
- oneflow/user/ops/unpack_op.cpp 2 additions, 2 deletionsoneflow/user/ops/unpack_op.cpp
Please register or sign in to comment