Searched defs:Im2col (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | im2col_utils.h | 213 void Im2col(const ConvParams& params, int kheight, int kwidth, uint8 zero_byte, in Im2col() function
248 void Im2col(const ConvParams& params, int kheight, int kwidth, in Im2col() function
|
| D | legacy_optimized_ops.h | 2394 void Im2col(const T* input_data, const Dims<4>& input_dims, int stride_width, in Im2col() function
2414 void Im2col(const T* input_data, const Dims<4>& input_dims, int stride, in Im2col() function
2834 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 | 73 void Im2col(const T* input_data, const int depth, const int height, in Im2col() function
|
| D | conv_grad_ops_3d.cc | 125 void Im2col(const T* input_data, const int depth, const int planes, in Im2col() function
|