| 62 | } |
| 63 | |
| 64 | size_t ConvPoolingForwardImpl::get_workspace_in_bytes( |
| 65 | const TensorLayout& src, const TensorLayout& filter, const TensorLayout& bias, |
| 66 | const TensorLayout& /*dst*/) { |
| 67 | // Worksapce contains the output of convolution layer in the workspace. |
| 68 | TensorLayout tmp_layout; |
| 69 | this->deduce_layout(src, filter, bias, tmp_layout); |
| 70 | return conv_dst_layout.total_nr_elems() * sizeof(float); |
| 71 | } |
| 72 | |
| 73 | void ConvPoolingForwardImpl::exec( |
| 74 | const _megdnn_in TensorND src, const _megdnn_in TensorND filter, |
nothing calls this directly
no test coverage detected