Searched refs:cpu_reducer_2d_t (Results 1 – 12 of 12) sorted by relevance
415 void cpu_reducer_2d_t<data_type>::conf_t::init_scratchpad( in init_scratchpad()420 * cpu_reducer_2d_t<data_type>::space_per_thread(balancer_); in init_scratchpad()427 cpu_reducer_2d_t<data_type>::cpu_reducer_2d_t(const conf_t &conf) in cpu_reducer_2d_t() function in dnnl::impl::cpu::aarch64::cpu_reducer_2d_t437 cpu_reducer_2d_t<data_type>::~cpu_reducer_2d_t() { in ~cpu_reducer_2d_t()442 status_t cpu_reducer_2d_t<data_type>::create_kernel() { in create_kernel()447 typename cpu_reducer_2d_t<data_type>::data_t *448 cpu_reducer_2d_t<data_type>::get_local_ptr( in get_local_ptr()458 int cpu_reducer_2d_t<data_type>::choose_x_blocking( in choose_x_blocking()502 void cpu_reducer_2d_t<data_type>::reduce_nolock(int ithr, data_t *dst, in reduce_nolock()563 template struct cpu_reducer_2d_t<data_type::f32>; variable[all …]
251 struct cpu_reducer_2d_t { struct275 cpu_reducer_2d_t(const conf_t &conf);276 ~cpu_reducer_2d_t();331 DNNL_DISALLOW_COPY_AND_ASSIGN(cpu_reducer_2d_t);
400 void cpu_reducer_2d_t<data_type>::conf_t::init_scratchpad( in init_scratchpad()405 * cpu_reducer_2d_t<data_type>::space_per_thread(balancer_); in init_scratchpad()412 cpu_reducer_2d_t<data_type>::cpu_reducer_2d_t(const conf_t &conf) in cpu_reducer_2d_t() function in dnnl::impl::cpu::x64::cpu_reducer_2d_t422 cpu_reducer_2d_t<data_type>::~cpu_reducer_2d_t() { in ~cpu_reducer_2d_t()427 status_t cpu_reducer_2d_t<data_type>::create_kernel() { in create_kernel()432 typename cpu_reducer_2d_t<data_type>::data_t *433 cpu_reducer_2d_t<data_type>::get_local_ptr( in get_local_ptr()443 int cpu_reducer_2d_t<data_type>::choose_x_blocking( in choose_x_blocking()487 void cpu_reducer_2d_t<data_type>::reduce_nolock(int ithr, data_t *dst, in reduce_nolock()548 template struct cpu_reducer_2d_t<data_type::f32>; variable[all …]
250 struct cpu_reducer_2d_t { struct274 cpu_reducer_2d_t(const conf_t &conf);275 ~cpu_reducer_2d_t();330 DNNL_DISALLOW_COPY_AND_ASSIGN(cpu_reducer_2d_t);
456 cpu_reducer_2d_t<data_type::f32>::conf_t reducer_wei_conf_;524 std::unique_ptr<cpu_reducer_2d_t<data_type::f32>> reducer_weights_;
496 new cpu_reducer_2d_t<data_type::f32>(pd()->reducer_wei_conf_))); in init()
452 cpu_reducer_2d_t<data_type::f32>::conf_t reducer_wei_conf_;520 std::unique_ptr<cpu_reducer_2d_t<data_type::f32>> reducer_weights_;