| 95 | return opr; |
| 96 | } |
| 97 | bool is_global_pooling_reduce(PoolingImpl::PoolingKernSizeParam& param) { |
| 98 | bool fmt_ok = param.format == PoolingImpl::Param::Format::NCHW || |
| 99 | param.format == PoolingImpl::Param::Format::NCHW44 || |
| 100 | param.format == PoolingImpl::Param::Format::NCHW88; |
| 101 | bool size_ok = param.filter[0] == param.isz[0] && param.filter[1] == param.isz[1] && |
| 102 | param.padding[0] == 0 && param.padding[1] == 0 && |
| 103 | param.osz[0] == 1 && param.osz[1] == 1; |
| 104 | bool dtype_ok = param.src_type == param.dst_type && |
| 105 | param.src_type.enumv() != DTypeEnum::Int8; |
| 106 | return fmt_ok && size_ok && dtype_ok; |
| 107 | } |
| 108 | |
| 109 | } // namespace |
| 110 | size_t PoolingImpl::get_workspace_in_bytes( |
no test coverage detected