Method Compute

oneflow/extension/python/py_kernel_caller.cpp:20–22

namespace oneflow {
void PyForwardKernel::Compute(user_op::KernelComputeContext* ctx) const {
  ::oneflow::pyext::PyCompute(ctx, "forward");
}

void PyBackwardKernel::Compute(user_op::KernelComputeContext* ctx) const {
  ::oneflow::pyext::PyCompute(ctx, "backward");