Searched refs:pool_width (Results 1 – 6 of 6) sorted by relevance
71 const float pool_width = pool_info.pooled_width();76 …t_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH)] / roi_scale) / pool_width);78 const auto min_width = static_cast<float>(pool_width / roi_scale);85 …std::uniform_int_distribution<> dist_w(min_width, std::max(float(min_width), (pool_width - 2) *…
72 const float pool_width = pool_info.pooled_width();77 …t_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH)] / roi_scale) / pool_width);79 const auto min_width = static_cast<float>(pool_width / roi_scale);86 …std::uniform_int_distribution<> dist_w(min_width, std::max(float(min_width), (pool_width - 2) *…
99 const int32_t pool_width = pool_rng(); in TEST() local107 .PoolingWidth(pool_width) in TEST()109 .StrideWidth(pool_width) in TEST()131 const int32_t pool_width = pool_rng(); in TEST() local139 .PoolingWidth(pool_width) in TEST()141 .StrideWidth(pool_width) in TEST()
101 const int32_t pool_width = pool_rng(); in TEST() local110 .PoolingWidth(pool_width) in TEST()112 .StrideWidth(pool_width) in TEST()134 const int32_t pool_width = pool_rng(); in TEST() local143 .PoolingWidth(pool_width) in TEST()145 .StrideWidth(pool_width) in TEST()