Searched refs:Reduce_Mean (Results 1 – 8 of 8) sorted by relevance
48 void Init(const bool keep_dims = false, const ReduceMode mode = ReduceMode::Reduce_Mean,
35 prim->set_mode(mindspore::ReduceMode::Reduce_Mean); in Parse()
46 prim->set_mode(mindspore::ReduceMode::Reduce_Mean); in Parse()
38 prim->set_mode(mindspore::ReduceMode::Reduce_Mean); in Parse()
51 prim->set_mode(mindspore::ReduceMode::Reduce_Mean); in Parse()
79 Reduce_Mean = 0, enumerator
75 mean2_primitive->get_mode() != mindspore::Reduce_Mean) { in IsReduceNode()
170 {ops::kNameReduceMax, mindspore::Reduce_Max}, {ops::kNameReduceMean, mindspore::Reduce_Mean},