Merge branch 'dev_bn_unimplement_bug' of...

Merge branch 'dev_bn_unimplement_bug' of https://github.com/Oneflow-Inc/oneflow into dev_xuan_act_event


Former-commit-id: 227a96e7ac56d1a5bbaa3eb138153460b6530dcb

oneflow/core/kernel/normalization_kernel.cpp

@@ -395,6 +395,12 @@ void NormalizationKernel<device_type, T>::UpdateMovingMeanAndMovingVariance(
       variance_blob->dptr<T>(), 1, moving_variance_blob->mut_dptr<T>(), 1);
 }
 
+template<DeviceType device_type, typename T>
+const PbMessage& NormalizationKernel<device_type, T>::GetCustomizedOpConf()
+    const {
+  return this->op_conf().normalization_conf();
+}
+
 ADD_DEFAULT_KERNEL_CREATOR(OperatorConf::kNormalizationConf,
                            NormalizationKernel, FLOATING_DATA_TYPE_SEQ);
 

oneflow/core/kernel/normalization_kernel.h

@@ -48,6 +48,7 @@ class NormalizationKernel final : public KernelIfWithActivation<device_type, T>,
                            const Blob* in_blob) const;
   void UpdateMovingMeanAndMovingVariance(
       const KernelCtx&, const std::function<Blob*(const std::string&)>&) const;
+  const PbMessage& GetCustomizedOpConf() const override;
 };
 
 }  // namespace oneflow