Home
last modified time | relevance | path

Searched refs:TRANSPOSE_B (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dcpu_kernel.h56 constexpr char TRANSPOSE_B[] = "transpose_b"; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/
Dmatmul_cpu_kernel.cc185 bool trans_b = AnfAlgo::GetNodeAttr<bool>(kernel_node, TRANSPOSE_B); in InitKernel()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dops_utils.h128 constexpr char TRANSPOSE_B[] = "transpose_b"; variable
Dmatmul_info.cc87 auto transpose_b_iter = attrs_.find(TRANSPOSE_B); in GetAttrs()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/
Drec_parse_graph.cc124 bool transpose_b = attrs[TRANSPOSE_B]->cast<BoolImmPtr>()->value(); in Complete2DInputs()
Drec_generate_strategy.cc70 bool transpose_b = attrs[TRANSPOSE_B]->cast<BoolImmPtr>()->value(); in PrepareMatMul()