Searched defs:shape_a (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ops/_op_impl/_custom_op/ |
| D | _basic.py | 24 def _get_km_kn_shape(shape_a, shape_b, trans_a, trans_b): argument
69 def _check_bias(shape_bias, shape_a, shape_b, m_shape, n_shape): argument
86 def _shape_check(shape_a, shape_b, shape_bias, src_dtype, trans_a, trans_b): argument
|
| D | matmul_cube_impl.py | 160 def _get_block(shape_a, shape_b, trans_a, trans_b): argument
184 def _get_format(input_x1, input_x2, trans_a, trans_b, block_in, block_out, shape_a, shape_b): argument
|
| D | matmul_cube_dense_left_impl.py | 99 def shape_gen2(bias, input_x1, output_y, shape_a, shape_b, trans_a, trans_b): argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | insert_pad.cc | 38 auto TransANotTransB = [](const vec &shape_a, const vec &shape_b, vec *pad_shape_a, vec *pad_shape_… in __anon052b6a880302()
53 auto TransATransB = [](const vec &shape_a, const vec &shape_b, vec *pad_shape_a, vec *pad_shape_b) { in __anon052b6a880402()
68 auto NotTransATransB = [](const vec &shape_a, const vec &shape_b, vec *pad_shape_a, vec *pad_shape_… in __anon052b6a880502()
83 auto NotTransANotTransB = [](const vec &shape_a, const vec &shape_b, vec *pad_shape_a, vec *pad_sha… in __anon052b6a880602()
114 auto shape_a = AnfAlgo::GetInputDeviceShape(matmul, 0); in NeedPad() local
|
| /third_party/mindspore/mindspore/nn/probability/distribution/_utils/ |
| D | utils.py | 242 def raise_broadcast_error(shape_a, shape_b): argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | common_utils.cc | 545 bool IsSameShape(const std::vector<size_t> &shape_a, const std::vector<size_t> &shape_b) { in IsSameShape()
|