Home
last modified time | relevance | path

Searched defs:matmul_split_inputs (Results 1 – 1 of 1) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/
Dinference_matmul_split_fusion.cc220 std::vector<AnfNodePtr> matmul_split_inputs = {input_x, input_w, tuple_node}; in CreateMatmulSplitNode() local
265 std::vector<AnfNodePtr> matmul_split_inputs = {matmul_x, matmul_w, reshape_tuple, input_bias}; in CreateMatmulBiasAddSplitNode() local
396 std::vector<AnfNodePtr> matmul_split_inputs = {x_node, weight_node, tuple}; in CreateMatmulSplitSiluNode() local
453 std::vector<AnfNodePtr> matmul_split_inputs = {matmul_input, input_w, tuple_node, input_bias}; in CreateMatmulBiasAddSplitSiluNode() local