| /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/ |
| drop_im2col_arrays.cc | 29 // Conv op does not have im2col.
33 // Drop the im2col array.
37 AddMessageF("Dropped an im2col array for %s", LogName(*conv_op));
|
| create_im2col_arrays.cc | 35 // We already have an im2col array
41 // from the weights dims we determine whether an im2col array is
50 // 1x1 unstrided conv does not need an im2col array.
54 // Create the im2col array.
61 "Created an im2col array for %s, with %dx%d kernel and stride_width=%d, "
|
| convert_pure_conv_to_depthwise.cc | 57 // Conv may have a 2nd output for im2col
60 // delete the im2col array.
|
| unfuse_activation_functions.cc | 32 // If a conv operation has an im2col array, yield: it should be dropped first.
|
| propagate_fixed_sizes.cc | 174 // Set im2col array dimensions if there is one.
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| conv.cc | 80 // Instead, we allocate a new object to use as scratch space for im2col, and
202 // We don't always need to allocate im2col. It is only used in some versions
243 TfLiteTensor* im2col = local
245 im2col->type = data_type;
246 im2col->allocation_type = kTfLiteArenaRw;
247 auto im2col_status = context->ResizeTensor(context, im2col, im2col_size);
294 TfLiteTensor* im2col, TfLiteTensor* hwcn_weights,
312 GetTensorDims(output), GetTensorData<uint8_t>(im2col),
313 GetTensorDims(im2col), gemm_context);
327 GetTensorDims(output), GetTensorData<uint8_t>(im2col),
407 TfLiteTensor* im2col = local
[all...] |
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| api_def_FusedPadConv2D.pbtxt | 40 im2col lookup, but in this case without the bilinear filtering required for
|
| /external/tensorflow/tensorflow/core/kernels/ |
| conv_ops_fused.cc | 252 // Combines bilinear resizing and mirror padding into the im2col transformation
320 // The im2col buffer has # of patches rows, and # of filters cols.
334 errors::InvalidArgument("Im2Col patch too large for buffer"));
353 // into filter patches by im2col.
363 // incrementally as we progress downwards through the input creating im2col
398 // mirrored inputs to the im2col operation. The problem is that we want to
402 // transformations into the im2col lookup itself. This successfully reduced
403 // memory usage, but because im2col can access an individual pixel for many
407 // horizontal slice of the resized and padded input to the im2col
447 // mirrored. This allows the following im2col code to access th
[all...] |
| conv_ops_using_gemm.cc | 206 // the input image into columns (im2col) and then running GEMM to produce the
236 // We can just use a GEMM if the im2col is the identity operator, e.g., if
286 // The im2col buffer has # of patches rows, and # of filters cols.
299 errors::InvalidArgument("Im2Col patch too large for buffer"));
355 // What we're doing here is trying to copy and fill the im2col
|
| quantized_conv_ops.cc | 195 // the input image into columns (im2col) and then running GEMM to produce the
246 // The im2col buffer has # of patches rows, and # of filters cols.
324 // What we're doing here is trying to copy and fill the im2col
|
| conv_grad_filter_ops.cc | 59 void Im2col(const T* input_data, const int depth, const int height,
470 // to do im2col to allow gemm-type computation.
471 Im2col<T>(
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| cblas_conv.h | 52 optimized_ops::Im2col(input_data, input_dims, stride_width, stride_height,
|
| optimized_ops.h | 722 void Im2col(const T* input_data, const Dims<4>& input_dims, int stride_width,
726 gemmlowp::ScopedProfilingLabel label("Im2col");
[all...] |
| /external/tensorflow/tensorflow/compiler/tests/ |
| extract_image_patches_op_test.py | 56 name="im2col")
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| extract_image_patches_op_test.py | 53 name="im2col")
|
| /external/gemmlowp/public/ |
| output_stages.h | 111 // are often implemented using 0-padding, or "im2col"-like
|
| /external/gemmlowp/doc/ |
| quantization_example.cc | 96 // which is required in multiple places, for example in Im2col with SAME
|
| /external/tensorflow/tensorflow/contrib/lite/toco/ |
| tooling_util.h | 196 // which is required in multiple places, for example in Im2col with SAME
|
| model.h | 349 // outputs[1]: optional: the intermediate array of im2col-replicated input
351 // of Conv layers as Im2col+GEMM.
[all...] |
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| README.md | 517 convolution's im2col patch generation. This transform looks out for that
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/reference/ |
| reference_ops.h | [all...] |
