Searched refs:need_im2col (Results 1 – 3 of 3) sorted by relevance
89 bool need_im2col; member157 data->need_im2col = in AllocateTemporaryTensorsIfRequired()175 if (data->need_im2col) { in AllocateTemporaryTensorsIfRequired()336 if (data->need_im2col) { in Prepare()673 data->need_im2col in Eval()
2139 const bool need_im2col = stride_width != 1 || stride_height != 1 || in Conv() local2146 } else if (need_im2col) { in Conv()2223 const bool need_im2col = stride_width != 1 || stride_height != 1 || in HybridConv() local2226 if (need_im2col) { in HybridConv()2388 const bool need_im2col = stride_width != 1 || stride_height != 1 || in Conv() local2399 } else if (need_im2col) { in Conv()
2208 const bool need_im2col = stride_width != 1 || stride_height != 1 || in Conv() local2215 } else if (need_im2col) { in Conv()2305 const bool need_im2col = stride_width != 1 || stride_height != 1 || in HybridConv() local2308 if (need_im2col) { in HybridConv()2391 const bool need_im2col = stride_width != 1 || stride_height != 1 || in Conv() local2402 } else if (need_im2col) { in Conv()