| /external/ComputeLibrary/src/core/NEON/kernels/convolution/winograd/ |
| D | output_transform.hpp | 68 TOut *outptr, size_t ld_out_row, size_t ld_out_col,
78 TOut *outptr, size_t ld_out_batch, size_t ld_out_row, size_t ld_out_col, in execute_internal() argument
110 auto outptr_row = outptr + thread_id*ld_out_row*this->get_output_rows(); in execute_internal()
122 outptr_row += n_threads * this->get_output_rows() * ld_out_row; in execute_internal()
133 outptr_tile, ld_out_row, ld_out_col, in execute_internal()
178 void *outptr, size_t ld_out_batch, size_t ld_out_row, size_t ld_out_col, in execute() argument
186 reinterpret_cast<TOut *>(outptr), ld_out_batch, ld_out_row, ld_out_col, in execute()
199 TOut *outptr, size_t ld_out_row, size_t ld_out_col,
216 TOut *outptr, size_t ld_out_row, size_t ld_out_col, in execute_tile() argument
224 auto kernel_ld_out_row = ld_out_row, kernel_ld_out_col = ld_out_col; in execute_tile()
[all …]
|
| D | weight_transform.hpp | 54 TOut *outptr, size_t ld_out_matrix, size_t ld_out_row, in execute_internal() argument
65 outptr += offset * ld_out_row; in execute_internal()
79 outptr += ld_out_row; in execute_internal()
86 outptr += skip * ld_out_row; in execute_internal()
115 void *outptr, size_t ld_out_matrix, size_t ld_out_row, in execute() argument
122 reinterpret_cast<TOut *>(outptr), ld_out_matrix, ld_out_row, in execute()
|
| D | input_transform.hpp | 85 TOut *outptr, size_t ld_out_batch, size_t ld_out_matrix, size_t ld_out_row, in execute_internal() argument
105 auto outptr_tile = outptr + thread_id * n_tile_cols * ld_out_row; in execute_internal()
133 outptr_tile += ld_out_row; in execute_internal()
136 outptr_tile += (n_threads - 1) * n_tile_cols * ld_out_row; in execute_internal()
163 void *outptr, size_t ld_out_batch, size_t ld_out_matrix, size_t ld_out_row, in execute() argument
170 reinterpret_cast<TOut *>(outptr), ld_out_batch, ld_out_matrix, ld_out_row, in execute()
|
| /external/ComputeLibrary/src/core/NEON/kernels/assembly/ |
| D | winograd.hpp | 123 void *outptr, size_t ld_out_matrix, size_t ld_out_row,
158 void *outptr, size_t ld_out_batch, size_t ld_out_matrix, size_t ld_out_row,
188 void *outptr, size_t ld_out_batch, size_t ld_out_row, size_t ld_out_col, in execute() argument
195 outptr, ld_out_batch, ld_out_row, ld_out_col, in execute()
203 void *outptr, size_t ld_out_batch, size_t ld_out_row, size_t ld_out_col,
|
| /external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/ |
| D | depthwise_planar.hpp | 305 TOutput *outptr, size_t ld_out_row, size_t ld_out_col, size_t ld_out_vl, in execute_kernel() argument
319 outptr += ld_out_row; in execute_kernel()
|
| D | depthwise_depthfirst.hpp | 248 TOutput *outptr, size_t ld_out_row, size_t ld_out_col, in direct()
255 outptr, ld_out_row, ld_out_col, in direct()
|