| 29 | } |
| 30 | |
| 31 | ElemwiseOpParamN<2> AddUpdateForwardHelper::make_param( |
| 32 | _megdnn_tensor_inout dst, _megdnn_tensor_in delta) { |
| 33 | ElemwiseOpParamN<2> src; |
| 34 | src[0] = dst; |
| 35 | src[1] = delta; |
| 36 | src[1].layout = src[1].layout.broadcast(dst.layout); |
| 37 | src.init_from_given_tensor(); |
| 38 | |
| 39 | return src; |
| 40 | } |
| 41 | } // namespace megdnn |
| 42 | |
| 43 | // vim: syntax=cpp.doxygen |
nothing calls this directly
no test coverage detected