Searched defs:batch_matmul (Results 1 – 3 of 3) sorted by relevance
30 auto batch_matmul = cnode->input(kIndex2); in MatchBatchMatmulFusedMulAdd() local
420 auto batch_matmul = func_graph->NewCNode(matmul_inputs); in CreateBatchMatMul() local439 auto batch_matmul = func_graph->NewCNode(matmul_inputs); in CreateBatchMatMul2() local452 const AnfNodePtr &batch_matmul) { in CreateDwReduceSum()469 const AnfNodePtr &batch_matmul) { in CreateDwReshape()504 auto batch_matmul = CreateBatchMatMul2(func_graph, lstm_input_grad, value_node); in CreateDbReduceSum() local557 auto batch_matmul = CreateBatchMatMul(func_graph, lstm_input_grad, concat); in Process() local
300 auto batch_matmul = func_graph->NewCNode(matmul_inputs); in AddDwhMatmulNode() local367 auto batch_matmul = graph->NewCNode(matmul_inputs); in CreateDwxBatchMatMul() local385 auto batch_matmul = func_graph->NewCNode(matmul_inputs); in CreateDxtBatchMatMul() local