• Y
    Polish OpWithKernel · bbdac7f7
    Yu Yang 提交于
    * Chage `IndicateDataType` to `GetKernelType`. Make it easier to
      understand.
    * Change `OpKernelKey` to `OpKernelType`
    * Make operator developers can customize which kernel the operator will
      use in runtime.
    bbdac7f7
softmax_with_cross_entropy_op.cc 8.3 KB