Searched refs:need_im2col (Results 1 – 4 of 4) sorted by relevance
59 const bool need_im2col = stride_width != 1 || stride_height != 1 || in ConvPerChannel() local71 } else if (need_im2col) { in ConvPerChannel()
108 bool need_im2col = false; member177 const bool need_im2col = need_dilated_im2col || need_non_dilated_im2col; in IsIm2ColRequired() local180 if (!need_im2col) return false; in IsIm2ColRequired()245 data->need_im2col = in AllocateTemporaryTensorsIfRequired()249 if (data->need_im2col) { in AllocateTemporaryTensorsIfRequired()453 if (data->need_im2col) { in Prepare()1002 data->need_im2col in EvalImpl()
1264 const bool need_im2col = stride_width != 1 || stride_height != 1 || in Conv() local1271 } else if (need_im2col) { in Conv()1362 const bool need_im2col = stride_width != 1 || stride_height != 1 || in HybridConv() local1370 } else if (need_im2col) { in HybridConv()1450 const bool need_im2col = stride_width != 1 || stride_height != 1 || in HybridConvPerChannel() local1462 } else if (need_im2col) { in HybridConvPerChannel()1565 const bool need_im2col = stride_width != 1 || stride_height != 1 || in Conv() local1576 } else if (need_im2col) { in Conv()
2450 const bool need_im2col = stride_width != 1 || stride_height != 1 || in Conv() local2457 } else if (need_im2col) { in Conv()2651 const bool need_im2col = stride_width != 1 || stride_height != 1 || in Conv() local2662 } else if (need_im2col) { in Conv()