• Y
    Add OperatorWithKernel class · 0ff81920
    Yu Yang 提交于
    * User can register OpKernel to its Ops. The OpKernelMap saved in
      OperatorWithKernel. Each Op which inherits OperatorWithKernel will
      use `OpKernel::Compute` instead of Run.
    0ff81920
op_registry_test.cc 5.4 KB