Searched refs:Col2im (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/ |
| D | transpose_conv.h | 90 optimized_ops::Col2im( in TransposeConvV2()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_input_ops.h | 77 void Col2im(const T* col_data, const int depth, const int height, in Col2im() function
705 Col2im<T>(
735 Col2im<T>(im2col_buf, dims.in_depth,
|
| D | conv_grad_ops_3d.cc | 74 void Col2im(const T* col_data, const int depth, const int planes, in Col2im() function
502 Col2im<T>(col_buffer_data, dims.in_depth, in Compute()
555 Col2im<T>(im2col_buf, dims.in_depth, in Compute()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 5662 void Col2im(const T* col_data, const int depth, const int height, in Col2im() function
5771 Col2im(col2im_data, output_depth, output_height, output_width, in TransposeConvV2()
6002 Col2im(col2im_data, output_depth, output_height, output_width, in TransposeConvV2()
|