| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/_op_impl/_custom_op/ |
| D | img2col_impl.py | 39 def shape56_0(tik_instance, input_x, res, input_shape, shape_info): argument
66 def shape56_1(tik_instance, input_x, res, input_shape, shape_info): argument
94 def shape56_2(tik_instance, input_x, res, input_shape, shape_info): argument
118 def shape56_3(tik_instance, input_x, res, input_shape, shape_info): argument
146 def shape56_4(tik_instance, input_x, res, input_shape, shape_info): argument
175 def shape28_0(tik_instance, input_x, res, input_shape, shape_info): argument
203 def shape28_1(tik_instance, input_x, res, input_shape, shape_info): argument
242 def shape28_2(tik_instance, input_x, res, input_shape, shape_info): argument
290 def shape28_3(tik_instance, input_x, res, input_shape, shape_info): argument
318 def shape28_4(tik_instance, input_x, res, input_shape, shape_info): argument
[all …]
|
| D | fused_abs_max1_impl.py | 37 def _update_tik(tik_instance, input_x_ub, broadcast_0_local_ub, block_index, res): argument
78 def shape0(tik_instance, input_x_shape, input_x, res): argument
102 def shape1(tik_instance, input_x_shape, ori_shape, input_x, res): argument
148 def shape2(tik_instance, input_x_shape, input_x, res): argument
173 def shape3_1000(tik_instance, input_x, res): argument
209 def shape3_1001(tik_instance, input_x, res): argument
244 def shape3(tik_instance, input_x_shape, ori_shape, input_x, res): argument
277 def shape4(tik_instance, input_x_shape, input_x, res): argument
304 def shape5(tik_instance, input_x_shape, input_x, res): argument
332 def shape6(tik_instance, ori_shape, input_x, res): argument
[all …]
|
| D | transpose02314_impl.py | 106 def shape0(tik_instance, input_x, res, dtype): argument
126 def shape1(tik_instance, input_x, res, dtype): argument
155 def shape2(tik_instance, input_x, res, dtype): argument
175 def shape3(tik_instance, input_x, res, dtype): argument
195 def shape4(tik_instance, input_x, res, dtype): argument
215 def shape5(tik_instance, input_x, res, dtype): argument
234 def shape6(tik_instance, input_x, res, dtype): argument
263 def shape7(tik_instance, input_x, res, dtype): argument
279 def shape8(tik_instance, input_x, res, dtype): argument
295 def shape9(tik_instance, input_x, res, dtype): argument
[all …]
|
| D | batch_matmul_impl.py | 66 def _inner_matmul_new(tik_instance, dtype, input_info, res, res_index): argument
99 def _inner_matmul_new_1_64_32_64(tik_instance, dtype, input_info, res, res_index): argument
119 def process_input_shape_640(input_shape, tik_instance, dtype, total_input, res): argument
170 def process_input_shape_1152(input_shape, tik_instance, dtype, total_input, res): argument
|
| D | matmul_cube_dense_right_impl.py | 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_fracz_right_mul_impl.py | 149 def cus_cube_matmul_right_mul(tik_instance, input_x1, input_x2, input_x3, argument
|
| D | matmul_cube_fracz_left_cast_impl.py | 228 def cus_cube_matmul_cast(tik_instance, input_x1, trans_a, input_x2, trans_b, argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/kernel_builder/ascend/tbe_and_aicpu/tbe/impl/ |
| D | matmul_tik.py | 40 def MK_TO_K1MK0(tik_instance, mk_input_tensor, k1mk0_tensor, dtype, k1, m, k0): argument
52 def KN_TO_K1NK0(tik_instance, kn_input_tensor, k1nk0_tensor, dtype, k1, n, k0): argument
69 def N1MN0_TO_MN(tik_instance, mn_output_tensor, n1mn0_tensor, dtype, n1, m, n0): argument
|