Home
last modified time | relevance | path

Searched refs:need_im2col (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Dcblas_conv.h48 const bool need_im2col = stride_width != 1 || stride_height != 1 || in Conv() local
50 if (need_im2col) { in Conv()
Doptimized_ops.h777 const bool need_im2col = stride_width != 1 || stride_height != 1 || in Conv() local
779 if (need_im2col) { in Conv()
858 const bool need_im2col = stride_width != 1 || stride_height != 1 || in Conv() local
860 if (need_im2col) { in Conv()
/external/tensorflow/tensorflow/contrib/lite/kernels/
Dconv.cc75 bool need_im2col; member
205 data->need_im2col = in Prepare()
220 if (data->need_im2col) { in Prepare()
232 if (data->need_im2col) { in Prepare()
408 data->need_im2col in Eval()