| 54 | } |
| 55 | |
| 56 | void ConvPoolingForwardImpl::deduce_layout( |
| 57 | const TensorLayout& srcl, const TensorLayout& filterl, |
| 58 | const TensorLayout& /*biasl*/, TensorLayout& dstl) { |
| 59 | setParamOfSublayers(); |
| 60 | convFwd->deduce_layout(srcl, filterl, conv_dst_layout); |
| 61 | poolFwd->deduce_layout(conv_dst_layout, dstl); |
| 62 | } |
| 63 | |
| 64 | size_t ConvPoolingForwardImpl::get_workspace_in_bytes( |
| 65 | const TensorLayout& src, const TensorLayout& filter, const TensorLayout& bias, |
no outgoing calls
no test coverage detected