| 849 | } |
| 850 | |
| 851 | virtual Tbias* GetBiasHandle(OpKernelContext* context, |
| 852 | std::shared_ptr<ConvFwdPd>& conv2d_fwd_pd, |
| 853 | const Tensor& bias_tensor) { |
| 854 | if (fuse_biasadd_) { |
| 855 | return static_cast<Tbias*>( |
| 856 | const_cast<Tbias*>(bias_tensor.flat<Tbias>().data())); |
| 857 | } |
| 858 | return nullptr; |
| 859 | } |
| 860 | |
| 861 | virtual void AllocateOutputTensor(OpKernelContext* context, |
| 862 | const ConvFwdPd& conv_prim_desc, |