Home
last modified time | relevance | path

Searched refs:ELEMWISE (Results 1 – 25 of 28) sorted by relevance

12

/third_party/mindspore/mindspore/_extends/graph_kernel/model/
Dmodel.py92 ELEMWISE = 2 variable in PrimLib
131 PrimLib.ELEMWISE if out_shape[i + delta] == in_shape[i] else PrimLib.BROADCAST)
160 'Add': Prim(ELEMWISE),
161 'Abs': Prim(ELEMWISE),
162 'Neg': Prim(ELEMWISE),
163 'Mul': Prim(ELEMWISE),
164 'Sub': Prim(ELEMWISE),
165 'Log': Prim(ELEMWISE),
166 'IsNan': Prim(ELEMWISE),
167 'IsInf': Prim(ELEMWISE),
[all …]
Dgraph_split.py152 if self.get_relation(to, idx) > PrimLib.ELEMWISE:
711 … if dom.pattern not in (PrimLib.ELEMWISE, PrimLib.BROADCAST) or len(dom.in_relations) != 1:
714 … if a.pattern > PrimLib.BROADCAST or len(a.out_relations) != 1 or r != PrimLib.ELEMWISE or \
720 if dom.pattern not in (PrimLib.ELEMWISE, PrimLib.BROADCAST):
724 … if a.pattern <= PrimLib.BROADCAST and r == PrimLib.ELEMWISE and a.check_acyclic(dom) and \
731 return dom.pattern > PrimLib.ELEMWISE or r > PrimLib.ELEMWISE
735 … if dom.pattern not in (PrimLib.ELEMWISE, PrimLib.BROADCAST) or len(dom.out_relations) != 1 or \
744 if dom.pattern not in (PrimLib.ELEMWISE, PrimLib.BROADCAST) or \
758 return a.pattern > PrimLib.ELEMWISE or r > PrimLib.REDUCE or r == PrimLib.BROADCAST
1003 … if dom.pattern not in (PrimLib.ELEMWISE, PrimLib.BROADCAST) or len(dom.in_relations) != 1:
[all …]
Dgraph_parallel.py134 if dom_type in (PrimLib.ELEMWISE, PrimLib.BROADCAST):
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/
Dfusion_base_pass.cc34 …AnfAlgo::GetFusionType(node) == kernel::FusionType::ELEMWISE && not_updatestate_nums == ELTWISE_US… in CheckEltWiseNode()
49 …AnfAlgo::GetFusionType(node) == kernel::FusionType::ELEMWISE && not_updatestate_nums == ELTWISE_US… in CheckDoubleInEltWiseNode()
64 …AnfAlgo::GetFusionType(node) == kernel::FusionType::ELEMWISE && not_updatestate_nums == ELTWISE_MU… in CheckMultiOutputEltWiseNode()
Dmatmul_eltwise_fusion_pass.cc55 AnfAlgo::GetFusionType(cnode) == kernel::FusionType::ELEMWISE && in MatchSingleFusionPattern()
63 AnfAlgo::GetFusionType(cnode) == kernel::FusionType::ELEMWISE && in MatchSingleFusionPattern()
Dbnupdate_eltwise_fusion_pass.cc59 AnfAlgo::GetFusionType(cnode) == kernel::FusionType::ELEMWISE && in MatchSingleFusionPattern()
Dconv_single_in_fusion_pass.cc69 …AnfAlgo::GetFusionType(cnode) == kernel::FusionType::ELEMWISE && cnode->inputs().size() == ELTWISE… in MatchSingleFusionPattern()
Deltwise_fusion_pass.cc68 …AnfAlgo::GetFusionType(cnode) == kernel::FusionType::ELEMWISE && cnode->inputs().size() == ELTWISE… in MatchSingleFusionPattern()
Dconv_double_in_fusion_pass.cc65 …AnfAlgo::GetFusionType(cnode) == kernel::FusionType::ELEMWISE && cnode->inputs().size() == ELTWISE… in MatchSingleFusionPattern()
Dmulti_output_fusion_pass.cc73 …AnfAlgo::GetFusionType(cnode) == kernel::FusionType::ELEMWISE && cnode->inputs().size() == ELTWISE… in MatchSingleFusionPattern()
Ddepthwiseconv_eltwise_fusion_pass.cc64 AnfAlgo::GetFusionType(cnode) == kernel::FusionType::ELEMWISE) { in MatchSingleFusionPattern()
Dbnupdate_eltwise_eltwise_fusion_pass.cc70 AnfAlgo::GetFusionType(cnode) == kernel::FusionType::ELEMWISE && in MatchSingleFusionPattern()
Dsegment_eltwise_fusion_pass.cc79 …AnfAlgo::GetFusionType(cnode) == kernel::FusionType::ELEMWISE && cnode->inputs().size() == ELTWISE… in MatchSingleFusionPattern()
Dconv2dbackprop_eltwise_eltwise_fusion_pass.cc81 AnfAlgo::GetFusionType(cnode) == kernel::FusionType::ELEMWISE && in MatchSingleFusionPattern()
Dreduce_eltwise_fusion_pass.cc82 …AnfAlgo::GetFusionType(cnode) == kernel::FusionType::ELEMWISE && cnode->inputs().size() == ELTWISE… in MatchSingleFusionPattern()
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/format_type/
Dinsert_cast_test.cc61 builder.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
69 builder1.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
123 builder1.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
/third_party/mindspore/tests/ut/cpp/pre_activate/pass/
Dallreduce_fusion_test.cc57 builder.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
98 builder.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
139 builder.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
196 builder.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
/third_party/mindspore/tests/ut/cpp/tbe/
Dtbe_json_creator_test.cc67 builder.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
110 builder.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
168 builder.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
222 builder.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
267 builder.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
339 builder.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
/third_party/mindspore/config/
Dop_info.config113 …outputs": [{"index": 0, "name": "output"}], "attr": [], "fusion_type": "ELEMWISE", "dtype_format":…
114 …outputs": [{"index": 0, "name": "output"}], "attr": [], "fusion_type": "ELEMWISE", "dtype_format":…
115 …outputs": [{"index": 0, "name": "output"}], "attr": [], "fusion_type": "ELEMWISE", "dtype_format":…
118 …outputs": [{"index": 0, "name": "output"}], "attr": [], "fusion_type": "ELEMWISE", "dtype_format":…
119 …outputs": [{"index": 0, "name": "output"}], "attr": [], "fusion_type": "ELEMWISE", "dtype_format":…
121 …outputs": [{"index": 0, "name": "output"}], "attr": [], "fusion_type": "ELEMWISE", "dtype_format":…
122 …outputs": [{"index": 0, "name": "output"}], "attr": [], "fusion_type": "ELEMWISE", "dtype_format":…
123 …ke_output", "param_type": "optional", "type": "bool"}], "fusion_type": "ELEMWISE", "dtype_format":…
124 …outputs": [{"index": 0, "name": "output"}], "attr": [], "fusion_type": "ELEMWISE", "dtype_format":…
125 …outputs": [{"index": 0, "name": "output"}], "attr": [], "fusion_type": "ELEMWISE", "dtype_format":…
[all …]
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/buffer_fusion/
Dbuffer_fusion_test.cc72 builder.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
143 builder.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
233 builder.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
315 builder.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fission/
Dtransdata_split_test.cc126 builder.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
174 builder.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/model/
Dop_node.h44 ELEMWISE, enumerator
80 …ElemwiseOp(const std::string &op, const std::string &node_name) : PrimOp(op, node_name, ELEMWISE) … in ElemwiseOp()
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fusion/
Dtranspose_transdata_fusion_test.cc107 builder.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dkernel.h50 ELEMWISE, enumerator
/third_party/mindspore/tests/ut/cpp/pre_activate/mem_reuse/
Dmem_reuse_test.cc129 relu_builder.SetFusionType(kernel::FusionType::ELEMWISE); in CreateKernelGraph()

12