Method ForwardDataContent

oneflow/core/kernel/identity_kernel.cpp:33–37

};

void IdentityKernel::ForwardDataContent(KernelContext* ctx) const {
  const Blob* in_blob = ctx->BnInOp2Blob("in");
  Blob* out_blob = ctx->BnInOp2Blob("out");
  AutoMemcpy(ctx->stream(), out_blob, in_blob);
}

void IdentityKernel::ForwardHeader(KernelContext* ctx) const {
  ctx->BnInOp2Blob("out")->CopyHeaderFrom(ctx->BnInOp2Blob("in"));