Home
last modified time | relevance | path

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

/third_party/mindspore/tests/ut/cpp/parallel/auto_parallel/
Dedge_costmodel_test.cc38 MatMulInfoPtr matmul1; member in mindspore::parallel::TestEdgeCostModel
Dgraph_costmodel_test.cc46 MatMulInfoPtr matmul1; member in mindspore::parallel::TestCostGraph
Ddp_algo_test.cc108 MatMulInfoPtr matmul1; member in mindspore::parallel::TestDPAlgo
/third_party/mindspore/mindspore/lite/tools/converter/parser/
Dtf_bidirection_gru_cf_fusion.cc71 auto matmul1 = VectorRef({GetPrim(prim::kPrimMatMul), concat1, matmul1_enter}); in DefineGruCellPattern() local
/third_party/mindspore/tests/ut/cpp/parallel/ops_info/
Dmatmul_info_test.cc32 MatMulInfoPtr matmul1; variable
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dtf_bidirection_gru_fusion.cc335 VectorRef matmul1 = VectorRef({is_var_matmul1, concat_input_h, placeholders[8]}); in GetBodyGraphPattern() local