Tf.extract_image_patches implementation

Question

Where is it sold tf.extract_image_patches? I checked the tensorflow repository which I could not find.

The file tensorflow/core/kernels/extract_image_patches_op.ccdoes not contain an implementation.

Answer 1

This is an interesting question. The path is as follows:

• The Python tf.extract_image_patchesfunction is implemented in a generated file tensorflow/python/ops/gen_array_ops.pythat calls "ExtractImagePatches"native op.

• This op is implemented ExtractImagePatchesOpin core/kernels/extract_image_patches_op.ccfor both processors and GPU devices. The main challenge functor::ExtractImagePatchesForward...

• ... which is defined in core/kernels/extract_image_patches_op.h. Actual logic is delegated to the ::extract_image_patches()input tensor method . Tensor type TTypes<T, 4>::ConstTensor, which is typedef Eigen::TensorMap core/framework/tensor_types.h.

, Eigen ( GitHub). , Eigen::TensorMap::extract_image_patches().

• eigen/unsupported/Eigen/CXX11/src/Tensor/TensorBase.h TensorImagePatchOp...

• ... eigen/unsupported/Eigen/CXX11/src/Tensor/TensorImagePatch.h. , , .

, Eigen tensorflow, bazel.