Searched refs:is_nchw (Results 1 – 6 of 6) sorted by relevance
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLGEMMDeconvolutionLayer.cpp | 42 …d_slice_coordinates(const ITensorInfo &output_info, const PadStrideInfo &deconv_info, bool is_nchw) in compute_start_end_slice_coordinates() argument
47 if(is_nchw) in compute_start_end_slice_coordinates()
133 const bool is_nchw = input->data_layout() == DataLayout::NCHW; in validate() local
146 if(is_nchw) in validate()
195 …nst auto start_end = compute_start_end_slice_coordinates(col2im_output_info, deconv_info, is_nchw); in validate()
202 …nst auto start_end = compute_start_end_slice_coordinates(col2im_output_info, deconv_info, is_nchw); in validate()
|
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NEDepthwiseConvolutionLayer.cpp | 206 bool is_nchw{ false }; member
235 _impl->is_nchw = input->info()->data_layout() == DataLayout::NCHW; in configure()
236 _impl->is_prepared = !_impl->is_nchw; in configure()
241 if(_impl->is_nchw) in configure()
261 if(_impl->is_nchw) in configure()
|
| /external/ComputeLibrary/src/core/NEON/kernels/ |
| D | NEBatchNormalizationLayerKernel.cpp | 343 const bool is_nchw = _input->info()->data_layout() == DataLayout::NCHW; in configure() local
344 if(is_nchw) in configure()
384 const bool is_nchw = _input->info()->data_layout() == DataLayout::NCHW; in run() local
385 if(is_nchw) in run()
|
| /external/ComputeLibrary/src/cpu/operators/ |
| D | CpuWinogradConv2d.cpp | 333 const bool is_nchw = _data_layout == DataLayout::NCHW; in run() local
334 if(is_nchw) in run()
345 …ITensorPack transform_input_pack{ { ACL_SRC, is_nchw ? input_nhwc.get() : src }, { ACL_DST, winogr… in run()
359 …utput_pack{ { ACL_SRC_0, winograd_output_transformed.get() }, { ACL_DST, is_nchw ? output_nhwc.get… in run()
361 if(is_nchw) in run()
|
| /external/ComputeLibrary/src/cpu/kernels/ |
| D | CpuIm2ColKernel.h | 110 template <typename T, bool has_pads, bool is_nchw>
|
| D | CpuIm2ColKernel.cpp | 251 template <typename T, bool has_pads, bool is_nchw>
293 if(is_nchw) in run_im2col()
|