Method InitOpKernelCacheWithFlags

oneflow/core/framework/op_kernel.h:300–303

  }

  virtual void InitOpKernelCacheWithFlags(KernelCacheContext* ctx, int8_t flag,
                                          std::shared_ptr<OpKernelCache>* cache_ptr) const {
    *cache_ptr = InitOpKernelCache(ctx);
  }

  virtual void Compute(KernelComputeContext* ctx, OpKernelState*, const OpKernelCache*) const {
    Compute(ctx);