| /third_party/mindspore/mindspore/ops/_op_impl/_custom_op/ |
| D | matmul_cube_fracz_right_mul_impl.py | 47 def cus_matmul_cube_fraczrightmul(input_x1, input_x2, input_x3, output_y=None, kernel_name="matmulc… argument
95 def get_cus_tile_info(input_x1, input_x2, input_x3): argument
142 def cus_cube_matmul_right_mul(tik_instance, input_x1, input_x2, input_x3, argument
|
| D | matmul_cube_dense_right_impl.py | 43 def cus_matmul_cube_dense_right(input_x1, input_x2, input_x3, output_y=None, argument
88 def _update_tik1(tik_instance, input_x1, input_x2, core_n_idx, resmatmul, matrix_max_scalar): argument
129 def _update_tik2(tik_instance, input_x1, input_x2, core_n_idx, resmatmul, matrix_max_scalar): argument
|
| D | matmul_cube_impl.py | 52 def cus_matmul_cube(input_x1, input_x2, bias=None, output_y=None, trans_a=False, trans_b=False, argument
127 def _get_shape_a_b(input_x1, input_x2, 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_fracz_left_cast_impl.py | 73 def cus_matmul_cube_fraczleftcast(input_x1, input_x2, bias=None, output_y=None, trans_a=False, tran… argument
190 def get_cus_tile_info(input_x1, input_x2, diag_size): argument
224 def cus_cube_matmul_cast(tik_instance, input_x1, trans_a, input_x2, trans_b, argument
|
| D | matmul_cube_dense_left_impl.py | 50 def shape_gen1(input_x1, input_x2, output_y, kernel_name, trans_a, trans_b): argument
214 def cus_matmul_cube_dense_left(input_x1, input_x2, bias=None, output_y=None, trans_a=False, trans_b… argument
|
| D | batch_matmul_impl.py | 189 def cus_batch_matmul(input_x1, input_x2, output, transpose_a=False, argument
|