Searched refs:kPooledH (Results 1 – 2 of 2) sorted by relevance
28 void ROIPooling::set_pooled_h(const int64_t pooled_h) { (void)this->AddAttr(kPooledH, MakeValue(poo… in set_pooled_h()30 int64_t ROIPooling::get_pooled_h() const { return GetValue<int64_t>(GetAttr(kPooledH)); } in get_pooled_h()64 auto new_h = GetValue<int64_t>(primitive->GetAttr(kPooledH)); in ROIPoolingInfer()
160 constexpr auto kPooledH = "pooled_h"; variable