Searched refs:out_batch_stride (Results 1 – 3 of 3) sorted by relevance
| /external/ComputeLibrary/src/core/NEON/kernels/convolution/common/ |
| D | shims.hpp | 60 int out_batch_stride=0,
93 int out_batch_stride=0,
152 int out_batch_stride, in nchw_to_nhwc() argument
168 out_batch_stride = (out_batch_stride) ? out_batch_stride in nchw_to_nhwc()
175 T* const out_batch = out + n*out_batch_stride; in nchw_to_nhwc()
272 int out_batch_stride, in nchw_to_nhwc() argument
282 out_batch_stride, out_row_stride, out_col_stride in nchw_to_nhwc()
297 int out_batch_stride, in nchw_to_nhwc() argument
307 out_batch_stride, out_row_stride, out_col_stride in nchw_to_nhwc()
325 int out_batch_stride, in nchw_to_nhwc() argument
[all …]
|
| /external/ComputeLibrary/src/cpu/kernels/ |
| D | CpuPermuteKernel.cpp | 197 const int out_batch_stride = dst->info()->strides_in_bytes()[3] / sizeof(T); in run_permute() local
204 out_batch_stride, out_row_stride, out_col_stride); in run_permute()
214 const int out_batch_stride = dst->info()->strides_in_bytes()[3] / sizeof(T); in run_permute() local
221 out_batch_stride, out_channel_stride, out_row_stride); in run_permute()
|
| D | CpuWinogradConv2dKernel.cpp | 90 const size_t out_batch_stride = dst_strides[batch_idx] / element_size_in_bytes; in run_op() local
106 out_batch_stride, in run_op()
|