| /third_party/mindspore/mindspore/core/ops/ |
| D | affine.cc | 22 …nit(const std::vector<int64_t> &contexts, int64_t output_dim, bool transpose_a, bool transpose_b) { in Init()
37 void Affine::set_transpose_b(bool transpose_b) { (void)AddAttr(kTransposeB, MakeValue(transpose_b))… in set_transpose_b()
|
| D | mat_mul.cc | 43 bool transpose_b = GetValue<bool>(transpose_b_ptr); in MatMulInferShape() local
103 void MatMul::Init(bool transpose_a, bool transpose_b) { in Init()
110 void MatMul::set_transpose_b(bool transpose_b) { (void)AddAttr(kTransposeB, MakeValue(transpose_b))… in set_transpose_b()
|
| D | batch_matmul.cc | 56 bool transpose_b = GetValue<bool>(transpose_b_ptr); in BatchMatmulInferShape() local
129 void BatchMatmul::Init(bool transpose_a, bool transpose_b) { in Init()
136 void BatchMatmul::set_transpose_b(bool transpose_b) { (void)AddAttr(kTransposeB, MakeValue(transpos… in set_transpose_b()
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_batch_matmul.py | 28 def __init__(self, transpose_a=False, transpose_b=False): argument
168 def __init__(self, transpose_a=False, transpose_b=False): argument
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_matmul.py | 27 def __init__(self, transpose_a=False, transpose_b=False): argument
|
| D | test_batch_matmul.py | 27 def __init__(self, transpose_a=False, transpose_b=False): argument
|
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_batch_matmul.py | 25 …def __init__(self, mul_weight, batch_matmul_weight, transpose_b=False, strategy1=None, strategy2=N… argument
|
| D | test_auto_star_elimination.py | 53 def __init__(self, transpose_a=False, transpose_b=False): argument
|
| /third_party/mindspore/tests/ut/cpp/parallel/ |
| D | step_auto_parallel_test.cc | 76 ValuePtr transpose_b = MakeValue(false); in Create_Node() local
118 ValuePtr transpose_b = MakeValue(false); in Create_two_nodes() local
|
| D | step_parallel_test.cc | 153 ValuePtr transpose_b = MakeValue(false); in Make_Manager() local
347 ValuePtr transpose_b = MakeValue(false); in TEST_F() local
502 ValuePtr transpose_b = MakeValue(false); in TEST_F() local
|
| /third_party/mindspore/mindspore/ops/_op_impl/_custom_op/ |
| D | batch_matmul_impl.py | 190 transpose_b=True, kernel_name="batchmatmul"): argument
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/ |
| D | rec_parse_graph.cc | 124 bool transpose_b = attrs[TRANSPOSE_B]->cast<BoolImmPtr>()->value(); in Complete2DInputs() local
|
| D | rec_generate_strategy.cc | 70 bool transpose_b = attrs[TRANSPOSE_B]->cast<BoolImmPtr>()->value(); in PrepareMatMul() local
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | prim_maths.cc | 286 bool transpose_b = GetValue<bool>(transpose_b_ptr); in InferImplMatMul() local
353 bool transpose_b = GetValue<bool>(transpose_b_ptr); in InferImplBatchMatMul() local
|
| /third_party/mindspore/tests/st/auto_parallel/ |
| D | parallel_strategy_search.py | 183 test_size, prelu_size, transpose_b, matmul_size, num_class): argument
|
| /third_party/mindspore/mindspore/ops/composite/ |
| D | math_ops.py | 671 def _infer_shape_rem(shape1, shape2, ndim1, ndim2, transpose_b): argument
|
| /third_party/mindspore/mindspore/parallel/nn/ |
| D | layers.py | 330 transpose_b=True, argument
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | matmul_info.cc | 35 …evMatrixShape(const Dimensions &mat_a_strategy, const Dimensions &mat_b_strategy, bool transpose_b, in SetDevMatrixShape()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/ |
| D | trt_op_converter.cc | 529 const auto &transpose_b = AnfAlgo::GetNodeAttr<bool>(node, "transpose_b"); in MS_TRT_CONVERTER_FUNC_REG() local
575 const auto &transpose_b = AnfAlgo::GetNodeAttr<bool>(node, "transpose_b"); in MS_TRT_CONVERTER_FUNC_REG() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/model/ |
| D | op_node.cc | 472 auto transpose_b = GetValue<bool>(attrs.find("transpose_b")->second); in InferShape() local
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | _thor_ops.py | 284 def __init__(self, transpose_a=False, transpose_b=False): argument
|
| D | math_ops.py | 1151 def __init__(self, transpose_a=False, transpose_b=False): argument
1256 def __init__(self, transpose_a=False, transpose_b=False): argument
|
| /third_party/mindspore/mindspore/numpy/ |
| D | math_ops.py | 3024 def _infer_shape_rem(shape1, shape2, ndim1, ndim2, transpose_b): argument
|
| /third_party/mindspore/mindspore/ccsrc/transform/express_ir/ |
| D | onnx_exporter.cc | 1064 auto transpose_b = GetValue<bool>(prim->GetAttr("transpose_b")); in ExportPrimBatchMatMul() local
|
| /third_party/mindspore/mindspore/lite/mindir/src/ |
| D | mindir.cc | 2093 PrimitivePtr MindIR_MatMulFusion_CreatePrimitive(bool transpose_a, bool transpose_b, ActivationType… in MindIR_MatMulFusion_CreatePrimitive()
2149 void MindIR_MatMulFusion_SetTransposeB(PrimitivePtr *primitive, bool transpose_b) { in MindIR_MatMulFusion_SetTransposeB()
|