Searched refs:im2col (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | transpose_conv.cc | 110 TfLiteTensor* im2col) { in ResizeIm2ColTensor() argument
126 im2col->type = input->type; in ResizeIm2ColTensor()
127 im2col->allocation_type = kTfLiteDynamic; in ResizeIm2ColTensor()
128 return context->ResizeTensor(context, im2col, im2col_shape_array); in ResizeIm2ColTensor()
146 TfLiteTensor* im2col = in Prepare() local
164 SetTensorToDynamic(im2col); in Prepare()
168 ResizeIm2ColTensor(context, output_shape, weights, input, im2col)); in Prepare()
182 TfLiteTensor* im2col = in Eval() local
192 if (IsDynamicTensor(im2col)) { in Eval()
194 weights, input, im2col)); in Eval()
[all …]
|
| D | conv.cc | 347 TfLiteTensor* im2col = in Prepare() local
349 im2col->type = input->type; in Prepare()
351 im2col->type = filter->type; in Prepare()
353 im2col->allocation_type = kTfLiteArenaRw; in Prepare()
354 auto im2col_status = context->ResizeTensor(context, im2col, im2col_size); in Prepare()
423 TfLiteTensor* im2col, TfLiteTensor* hwcn_weights, in EvalQuantized() argument
465 GetTensorShape(im2col), GetTensorData<uint8_t>(im2col), gemm_context); in EvalQuantized()
492 GetTensorShape(im2col), GetTensorData<uint8_t>(im2col), gemm_context); in EvalQuantized()
524 TfLiteTensor* filter, TfLiteTensor* bias, TfLiteTensor* im2col, in EvalFloat() argument
555 GetTensorData<float>(output), GetTensorShape(im2col), in EvalFloat()
[all …]
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_FusedPadConv2D.pbtxt | 40 im2col lookup, but in this case without the bilinear filtering required for
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | README.md | 518 convolution's im2col patch generation. This transform looks out for that
|