Skip to content
Snippets Groups Projects

Unified interface for attributes

Closed Olivier BICHLER requested to merge unified_params into master
1 file
+ 1
1
Compare changes
  • Side-by-side
  • Inline
@@ -47,7 +47,7 @@ void AvgPoolingImpl2D_cpu_forward_kernel(const AvgPooling_Op<2>::Attrs &attrs,
static_cast<float>(std::get<0>(attrs)[0])));
// output W size
const std::size_t oySize =
static_cast<std::size_t>(std::floor(static_cast<float>(dims[3] + std::get<2>(attrs)[1] + std::get<2>(attrs)[3] - std::get<1>(attrs)[1] + std::get<0>(attrs)/
static_cast<std::size_t>(std::floor(static_cast<float>(dims[3] + std::get<2>(attrs)[1] + std::get<2>(attrs)[3] - std::get<1>(attrs)[1] + std::get<0>(attrs)[1])/
static_cast<float>(std::get<0>(attrs)[1])));
// TODO: kernel computation
Loading