Searched defs:reduce_to_end (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/ |
| D | reduce_fp32_tests.cc | 72 … float *output_data, ReduceMode mode, int *axes, const int num_axes, bool reduce_to_end, in Prepare()
119 bool reduce_to_end = false; in TEST_F() local
146 bool reduce_to_end = true; in TEST_F() local
174 bool reduce_to_end = false; in TEST_F() local
200 bool reduce_to_end = false; in TEST_F() local
227 bool reduce_to_end = false; in TEST_F() local
256 bool reduce_to_end = true; in TEST_F() local
284 bool reduce_to_end = false; in TEST_F() local
309 bool reduce_to_end = true; in TEST_F() local
336 bool reduce_to_end = false; in TEST_F() local
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/fusion/ |
| D | reduce_fusion.cc | 39 void ReduceFusion::set_reduce_to_end(const bool reduce_to_end) { in set_reduce_to_end()
69 void ReduceFusion::Init(const bool keep_dims, const ReduceMode mode, const bool reduce_to_end, cons… in Init()
94 bool reduce_to_end = reduce_to_end_ptr && GetValue<bool>(reduce_to_end_ptr); in ReduceFusionInferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/common/ops/operator_populate/ |
| D | reduce_populate.cc | 42 auto reduce_to_end = base_operator->GetPrim()->GetAttr(mindspore::ops::kReduceToEnd); in PopulateReduceOpParameter() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/ |
| D | reshape_reduce_fusion.cc | 183 auto reduce_to_end = attr != nullptr && GetValue<bool>(attr); in CheckReduce() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/ |
| D | preprocess_dynamic_shape.cc | 306 …bool reduce_to_end = prim->GetAttr(ops::kReduceToEnd) != nullptr && GetValue<bool>(prim->GetAttr(o… in ReduceInferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/mindir/src/ |
| D | mindir_2.cc | 212 …td::vector<int64_t> &axes, bool keep_dims, ReduceMode reduce_mode, bool reduce_to_end, float coeff… in MindIR_GenOP_CreatePrimitive()
1220 void MindIR_GenOP_SetReduceToEnd(PrimitivePtr *primitive, bool reduce_to_end) { in MindIR_GenOP_SetReduceToEnd()
|
| D | mindir.cc | 3077 …ndIR_ReduceFusion_CreatePrimitive(bool keep_dims, ReduceMode mode, bool reduce_to_end, float coeff… in MindIR_ReduceFusion_CreatePrimitive()
3166 void MindIR_ReduceFusion_SetReduceToEnd(PrimitivePtr *primitive, bool reduce_to_end) { in MindIR_ReduceFusion_SetReduceToEnd()
|