Searched defs:n_batches (Results 1 – 12 of 12) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/convolution/common/ |
| H A D | shims.hpp | 145 const int n_batches, in nchw_to_nhwc()
265 const int n_batches, in nchw_to_nhwc()
290 const int n_batches, in nchw_to_nhwc()
318 const int n_batches, in nchw_to_nhwc()
375 const int n_batches, in nhwc_to_nchw()
505 const int n_batches, in nhwc_to_nchw()
531 const int n_batches, in nhwc_to_nchw()
560 const int n_batches, in nhwc_to_nchw()
|
| H A D | tensor.hpp | 39 int n_batches, n_rows, n_cols, n_channels; member
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/internal/ |
| H A D | CpuPool2dAssemblyWrapperKernel.cpp | 207 const unsigned int n_batches = src->dimension(idx_batches); in create_arm_pooling() local
246 const unsigned int n_batches = src->dimension(idx_batches); in create_arm_pooling_requant() local
|
| H A D | CpuDepthwiseConv2dAssemblyWrapperKernel.cpp | 67 const unsigned int n_batches = src->dimension(idx_batches); in create_arm_dwc() local
108 const unsigned int n_batches = src->dimension(idx_batches); in create_arm_dwc_quant() local
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/ |
| H A D | CpuPermuteKernel.cpp | 158 int n_batches = 0; in run_permute() local
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/pooling/ |
| H A D | depthfirst_driver.hpp | 129 unsigned int n_batches, in execute_internal()
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/assembly/ |
| H A D | depthwise.hpp | 52 unsigned int n_batches, input_rows, input_cols, input_channels; member
|
| H A D | pooling.hpp | 53 unsigned int n_batches, input_rows, input_cols, n_channels; member
|
| H A D | winograd.hpp | 39 unsigned int n_batches; member
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/ |
| H A D | depthfirst_driver.hpp | 145 unsigned int n_batches, in execute_internal()
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/operators/ |
| H A D | CpuWinogradConv2d.cpp | 201 const uint32_t n_batches = _winograd_impl.gemm_args->_nbatches; in configure() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | TensorConversions.cpp | 1274 const auto n_batches = compressed_indices.dim() - 1; in sparse_compressed_to_flipped() local
|