Searched refs:kIsTraining (Results 1 – 4 of 4) sorted by relevance
39 (void)this->AddAttr(kIsTraining, MakeValue(is_training)); in set_is_training()43 auto value_ptr = this->GetAttr(kIsTraining); in get_is_training()
34 void BatchNorm::set_is_training(const bool is_training) { (void)this->AddAttr(kIsTraining, MakeValu… in set_is_training()57 auto value_ptr = GetAttr(kIsTraining); in get_is_training()103 if (!GetValue<bool>(primitive->GetAttr(kIsTraining))) { in BatchNormInfer()
103 constexpr auto kIsTraining = "is_training"; variable
30 src_prim->AddAttr(ops::kIsTraining, MakeValue(false)); in Mapper()