Searched refs:reduce_prim (Results 1 – 2 of 2) sorted by relevance
22 auto reduce_prim = primitive->value_as_ReduceFusion(); in IsSupport() local23 if (reduce_prim == nullptr) { in IsSupport()27 reduce_mode_ = reduce_prim->mode(); in IsSupport()29 …MS_LOG(WARNING) << "Npu does not support reduce mode " << reduce_prim->mode() << " for op " << nam… in IsSupport()32 if (reduce_prim->reduce_to_end()) { in IsSupport()41 auto reduce_prim = primitive->value_as_ReduceFusion(); in Init() local42 if (reduce_prim == nullptr) { in Init()52 reduce_mean->set_attr_keep_dims(reduce_prim->keep_dims()); in Init()55 …MS_LOG(ERROR) << "Npu does not support reduce mode " << reduce_prim->mode() << " for op " << name_; in Init()
27 auto reduce_prim = primitive->value_as_Reduce(); in PopulateReduceParameter() local28 if (reduce_prim == nullptr) { in PopulateReduceParameter()40 reduce_param->keep_dims_ = static_cast<bool>(reduce_prim->keepDims()); in PopulateReduceParameter()41 reduce_param->reduce_to_end_ = reduce_prim->reduceToEnd(); in PopulateReduceParameter()42 reduce_param->coeff = reduce_prim->coeff(); in PopulateReduceParameter()43 auto axisVector = reduce_prim->axes(); in PopulateReduceParameter()59 reduce_param->mode_ = static_cast<int>(reduce_prim->mode()); in PopulateReduceParameter()