Home
last modified time | relevance | path

Searched refs:transpose_a (Results 1 – 25 of 50) sorted by relevance

12

/third_party/mindspore/mindspore/core/ops/
Dmat_mul.cc42 bool transpose_a = GetValue<bool>(transpose_a_ptr); in MatMulInferShape() local
44 int64_t x_col = x_last[static_cast<size_t>(!transpose_a)]; in MatMulInferShape()
51 << "(transpose_a=" << transpose_a << "})" in MatMulInferShape()
71 auto x_c = x_shp[(transpose_a ? 0 : 1)]; in MatMulInferShape()
81 auto make_shape = [&transpose_a, &transpose_b](ShapeVector &output, const ShapeVector xshp, in MatMulInferShape()
83 output.push_back(xshp[(transpose_a ? 1 : 0)]); in MatMulInferShape()
103 void MatMul::Init(bool transpose_a, bool transpose_b) { in Init() argument
104 set_transpose_a(transpose_a); in Init()
108 void MatMul::set_transpose_a(bool transpose_a) { (void)AddAttr(kTransposeA, MakeValue(transpose_a))… in set_transpose_a() argument
Dbatch_matmul.cc55 bool transpose_a = GetValue<bool>(transpose_a_ptr); in BatchMatmulInferShape() local
57 int64_t x_col = x_last[static_cast<size_t>(!transpose_a)]; in BatchMatmulInferShape()
64 << "(transpose_a=" << transpose_a << "})" in BatchMatmulInferShape()
85 auto x_c = x_shp[x_offset + (transpose_a ? 0 : 1)]; in BatchMatmulInferShape()
95 …auto make_shape = [&transpose_a, &transpose_b, &offset](ShapeVector &output, const ShapeVector xsh… in BatchMatmulInferShape()
109 output.push_back(xshp[x_offset + (transpose_a ? 1 : 0)]); in BatchMatmulInferShape()
129 void BatchMatmul::Init(bool transpose_a, bool transpose_b) { in Init() argument
130 set_transpose_a(transpose_a); in Init()
134 void BatchMatmul::set_transpose_a(bool transpose_a) { (void)AddAttr(kTransposeA, MakeValue(transpos… in set_transpose_a() argument
Daffine.cc22 void Affine::Init(const std::vector<int64_t> &contexts, int64_t output_dim, bool transpose_a, bool … in Init() argument
25 this->set_transpose_a(transpose_a); in Init()
35 void Affine::set_transpose_a(bool transpose_a) { (void)AddAttr(kTransposeA, MakeValue(transpose_a))… in set_transpose_a() argument
Dmat_mul.h40 void Init(bool transpose_a = false, bool transpose_b = false);
42 void set_transpose_a(bool transpose_a);
Dbatch_matmul.h37 void Init(bool transpose_a = false, bool transpose_b = false);
39 void set_transpose_a(bool transpose_a);
Daffine.h41 void Init(const std::vector<int64_t> &contexts, int64_t output_dim, bool transpose_a = false,
54 void set_transpose_a(bool transpose_a);
/third_party/mindspore/tests/st/ops/gpu/
Dtest_batch_matmul.py28 def __init__(self, transpose_a=False, transpose_b=False): argument
30 self.batch_matmul = P.BatchMatMul(transpose_a, transpose_b)
87 net = BatchMatMulNet(transpose_a=True)
131 net = BatchMatMulNet(transpose_a=True, transpose_b=True)
168 def __init__(self, transpose_a=False, transpose_b=False): argument
170 self.batch_matmul = P.BatchMatMul(transpose_a, transpose_b)
/third_party/mindspore/tests/st/ops/cpu/
Dtest_batch_matmul.py27 def __init__(self, transpose_a=False, transpose_b=False): argument
29 self.batch_matmul = P.BatchMatMul(transpose_a, transpose_b)
95 net = BatchMatMulNet(transpose_a=True)
145 net = BatchMatMulNet(transpose_a=True, transpose_b=True)
Dtest_matmul.py27 def __init__(self, transpose_a=False, transpose_b=False): argument
29 self.matmul = P.MatMul(transpose_a, transpose_b)
78 net = MatMulNet(transpose_a=True)
108 net = MatMulNet(transpose_a=True, transpose_b=True)
/third_party/mindspore/mindspore/_extends/graph_kernel/expanders/
Dmatmul.py29 self.transpose_a = self.attrs['transpose_a']
40 k_a = self.shape_a[-2] if self.transpose_a else self.shape_a[-1]
62 if self.transpose_a:
/third_party/mindspore/tests/ut/cpp/parallel/
Dstep_auto_parallel_test.cc75 ValuePtr transpose_a = MakeValue(false); in Create_Node() local
77 prim->set_attr("transpose_a", transpose_a); in Create_Node()
117 ValuePtr transpose_a = MakeValue(false); in Create_two_nodes() local
119 prim->set_attr("transpose_a", transpose_a); in Create_two_nodes()
131 prim2->set_attr("transpose_a", transpose_a); in Create_two_nodes()
Dstep_parallel_test.cc152 ValuePtr transpose_a = MakeValue(false); in Make_Manager() local
154 prim1->AddAttr("transpose_a", transpose_a); in Make_Manager()
176 prim2->AddAttr("transpose_a", transpose_a); in Make_Manager()
346 ValuePtr transpose_a = MakeValue(false); in TEST_F() local
348 prim->set_attr("transpose_a", transpose_a); in TEST_F()
501 ValuePtr transpose_a = MakeValue(false); in TEST_F() local
503 prim->set_attr("transpose_a", transpose_a); in TEST_F()
/third_party/mindspore/tests/ut/python/parallel/
Dtest_auto_star_elimination.py53 def __init__(self, transpose_a=False, transpose_b=False): argument
55 self.fc = P.MatMul(transpose_a=transpose_a, transpose_b=transpose_b)
Dtest_loss_and_optimizer.py49 self.matmul = P.MatMul(transpose_a=False, transpose_b=True).shard(strategy1)
83 self.matmul = P.MatMul(transpose_a=False, transpose_b=True).shard(strategy1)
117 self.matmul = P.MatMul(transpose_a=False, transpose_b=True).shard(strategy1)
152 self.matmul = P.MatMul(transpose_a=False, transpose_b=True).shard(strategy1)
188 self.matmul = P.MatMul(transpose_a=False, transpose_b=True).shard(strategy1)
Dtest_optimizer_clone_weight.py48 self.matmul = P.MatMul(transpose_a=False, transpose_b=True).shard(strategy1)
83 self.matmul = P.MatMul(transpose_a=False, transpose_b=True).shard(strategy1)
Dtest_parameter_init.py40 self.matmul = P.MatMul(transpose_a=False, transpose_b=True).shard(strategy1)
Dtest_get_parameter_layout.py30 self.matmul = P.MatMul(transpose_a=False, transpose_b=True).shard(strategy1)
/third_party/mindspore/mindspore/core/abstract/
Dprim_maths.cc285 bool transpose_a = GetValue<bool>(transpose_a_ptr); in InferImplMatMul() local
298 auto x_col = x_shp[(transpose_a ? 0 : 1)]; in InferImplMatMul()
308 auto make_shape = [&transpose_a, &transpose_b](ShapeVector &output, const ShapeVector xshp, in InferImplMatMul()
310 output.push_back(xshp[(transpose_a ? 1 : 0)]); in InferImplMatMul()
352 bool transpose_a = GetValue<bool>(transpose_a_ptr); in InferImplBatchMatMul() local
366 auto x_col = x_shp[offset + (transpose_a ? 0 : 1)]; in InferImplBatchMatMul()
376 auto make_shape = [&transpose_a, &transpose_b](ShapeVector &output, const ShapeVector xshp, in InferImplBatchMatMul()
390 output.push_back(xshp[offset + (transpose_a ? 1 : 0)]); in InferImplBatchMatMul()
/third_party/mindspore/tests/st/ops/graph_kernel/
Dtest_matmul_to_mul.py26 self.matmul = P.MatMul(transpose_a=False, transpose_b=False)
34 self.bmm = P.BatchMatMul(transpose_a=False, transpose_b=False)
Dtest_matmul.py26 self.matmul = P.MatMul(transpose_a=True, transpose_b=True)
34 self.matmul = P.MatMul(transpose_a=True, transpose_b=True)
/third_party/mindspore/tests/ut/python/ops/
Dtest_ops_check.py195 'block': P.MatMul(transpose_a=True),
201 'block': P.MatMul(transpose_a=True, transpose_b=True),
258 'block': (P.MatMul(transpose_a=True), {'exception': ValueError}),
264 'block': (P.MatMul(transpose_a=True, transpose_b=True), {'exception': ValueError}),
/third_party/mindspore/mindspore/lite/src/ops/populate/
Dmatmul_populate.cc36 param->a_transpose_ = value->transpose_a(); in PopulateMatMulParameter()
Daffine_populate.cc56 matmul_param->a_transpose_ = value->transpose_a(); in PopulateAffineParameter()
/third_party/mindspore/mindspore/ops/operations/
D_thor_ops.py284 def __init__(self, transpose_a=False, transpose_b=False): argument
287 self.transpose_a = transpose_a
292 if self.transpose_a:
/third_party/mindspore/mindspore/nn/layer/
Dthor_layer.py134 self.cube_matmul = P.MatMul(transpose_a=True)
140 self.cube_matmul = P.CusMatMulCube(transpose_a=True)
416 self.cube_matmul = P.CusMatMulCube(transpose_a=True)
607 self.cube_matmul = P.CusMatMulCube(transpose_a=True)
609 self.cube_matmul = P.MatMul(transpose_a=True)
823 self.cube_matmul = P.MatMul(transpose_a=True)

12