Searched refs:kRoundMode (Results 1 – 5 of 5) sorted by relevance
105 auto run_mode_val = dst_prim->GetAttr(ops::kRoundMode); in AdjustCaffePoolAttr()108 dst_prim->set_attr(ops::kRoundMode, MakeValue(run_mode_ge)); in AdjustCaffePoolAttr()125 auto run_mode_val = dst_prim->GetAttr(ops::kRoundMode); in AdjustOnnxPoolAttr()
85 (void)this->AddAttr(kRoundMode, MakeValue(swi)); in set_round_mode()89 auto value_ptr = GetAttr(kRoundMode); in get_round_mode()
63 (void)this->AddAttr(kRoundMode, MakeValue(swi)); in set_round_mode()67 auto value_ptr = GetAttr(kRoundMode); in get_round_mode()
172 constexpr auto kRoundMode = "round_mode"; variable