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/ |
| D | inference_matmul_split_fusion.cc | 220 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
|