Searched refs:kPooledW (Results 1 – 2 of 2) sorted by relevance
32 void ROIPooling::set_pooled_w(const int64_t pooled_w) { (void)this->AddAttr(kPooledW, MakeValue(poo… in set_pooled_w()35 auto value_ptr = GetAttr(kPooledW); in get_pooled_w()65 auto new_w = GetValue<int64_t>(primitive->GetAttr(kPooledW)); in ROIPoolingInfer()
161 constexpr auto kPooledW = "pooled_w"; variable