Searched defs:reduce_to_end (Results 1 – 3 of 3) sorted by relevance
72 bool reduce_to_end, float coeff) { in Prepare()115 bool reduce_to_end = false; in TEST_F() local142 bool reduce_to_end = true; in TEST_F() local170 bool reduce_to_end = false; in TEST_F() local196 bool reduce_to_end = false; in TEST_F() local223 bool reduce_to_end = false; in TEST_F() local252 bool reduce_to_end = true; in TEST_F() local280 bool reduce_to_end = false; in TEST_F() local305 bool reduce_to_end = true; in TEST_F() local332 bool reduce_to_end = false; in TEST_F() local[all …]
36 void ReduceFusion::set_reduce_to_end(const bool reduce_to_end) { in set_reduce_to_end()66 void ReduceFusion::Init(const bool keep_dims, const ReduceMode mode, const bool reduce_to_end, cons… in Init()
2886 …ndIR_ReduceFusion_CreatePrimitive(bool keep_dims, ReduceMode mode, bool reduce_to_end, float coeff… in MindIR_ReduceFusion_CreatePrimitive()2975 void MindIR_ReduceFusion_SetReduceToEnd(PrimitivePtr *primitive, bool reduce_to_end) { in MindIR_ReduceFusion_SetReduceToEnd()