Searched refs:_is_global_pooling_layer (Results 1 – 2 of 2) sorted by relevance
43 _is_global_pooling_layer(false), in CpuPool2d()64 …_is_global_pooling_layer = (src->dimension(idx_width) == pool_info.pool_size.width) && (src->… in configure()109 const auto hints = (_is_global_pooling_layer) ? Window::DimX : Window::DimY; in run()117 …NEScheduler::get().schedule_op(_pooling_layer_kernel.get(), _is_global_pooling_layer ? Window::Dim… in run()
78 bool _is_global_pooling_layer; variable