Searched refs:need_im2colData (Results 1 – 1 of 1) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | Conv2D.cpp | 208 const bool need_im2colData = needim2colData(filterShape, stride_width, stride_height, in convNhwc() local
216 convertShapeToDims(outputShape), need_im2colData ? im2colData : nullptr, im2colDim); in convNhwc()
258 const bool need_im2colData = needim2colData(filterShape, stride_width, stride_height, in convNhwc() local
268 need_im2colData ? im2colData : nullptr, im2colDim, &gemm_context); in convNhwc()
|