Searched defs:Im2col (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | im2col_utils.h | 214 void Im2col(const ConvParams& params, int kheight, int kwidth, uint8 zero_byte, in Im2col() function
249 void Im2col(const ConvParams& params, int kheight, int kwidth, in Im2col() function
|
| D | legacy_optimized_ops.h | 2396 void Im2col(const T* input_data, const Dims<4>& input_dims, int stride_width, in Im2col() function
2416 void Im2col(const T* input_data, const Dims<4>& input_dims, int stride, in Im2col() function
2836 void Im2col(const T* input_data, const Dims<4>& input_dims, int stride, in Im2col() function
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_filter_ops.cc | 74 void Im2col(const T* input_data, const int depth, const int height, in Im2col() function
|
| D | conv_grad_ops_3d.cc | 127 void Im2col(const T* input_data, const int depth, const int planes, in Im2col() function
|