FusedResizeAndPadConv2d

public final class FusedResizeAndPadConv2d

Performs a resize and padding as a preprocess during a convolution.

It's often possible to do spatial transformations more efficiently as part of the packing stage of a convolution, so this op allows for an optimized implementation where these stages are fused together. This prevents the need to write out the intermediate results as whole tensors, reducing memory pressure, and we can get some latency gains by merging the transformation calculations. The data_format attribute for Conv2D isn't supported by this op, and defaults to 'NHWC' order. Internally this op uses a single per-graph scratch buffer, which means that it will block if multiple versions are being run in parallel. This is because this operator is primarily an optimization to minimize memory usage.

Nested Classes

class FusedResizeAndPadConv2d.Options Optional attributes for FusedResizeAndPadConv2d  

Constants

String OP_NAME The name of this op, as known by TensorFlow core engine

Public Methods

Output<T>
asOutput()
Returns the symbolic handle of the tensor.
static <T extends TNumber> FusedResizeAndPadConv2d<T>
create(Scope scope, Operand<T> input, Operand<TInt32> size, Operand<TInt32> paddings, Operand<T> filter, String mode, List<Long> strides, String padding, Options... options)
Factory method to create a class wrapping a new FusedResizeAndPadConv2d operation.
Output<T>
output()
static FusedResizeAndPadConv2d.Options
resizeAlignCorners(Boolean resizeAlignCorners)

Inherited Methods

Constants

public static final String OP_NAME

The name of this op, as known by TensorFlow core engine

Constant Value: "FusedResizeAndPadConv2D"

Public Methods

public Output<T> asOutput ()

Returns the symbolic handle of the tensor.

Inputs to TensorFlow operations are outputs of another TensorFlow operation. This method is used to obtain a symbolic handle that represents the computation of the input.

public static FusedResizeAndPadConv2d<T> create (Scope scope, Operand<T> input, Operand<TInt32> size, Operand<TInt32> paddings, Operand<T> filter, String mode, List<Long> strides, String padding, Options... options)

Factory method to create a class wrapping a new FusedResizeAndPadConv2d operation.

Parameters
scope current scope
input 4-D with shape `[batch, in_height, in_width, in_channels]`.
size A 1-D int32 Tensor of 2 elements: `new_height, new_width`. The new size for the images.
paddings A two-column matrix specifying the padding sizes. The number of rows must be the same as the rank of `input`.
filter 4-D with shape `[filter_height, filter_width, in_channels, out_channels]`.
strides 1-D of length 4. The stride of the sliding window for each dimension of `input`. Must be in the same order as the dimension specified with format.
padding The type of padding algorithm to use.
options carries optional attributes values
Returns
  • a new instance of FusedResizeAndPadConv2d

public Output<T> output ()

public static FusedResizeAndPadConv2d.Options resizeAlignCorners (Boolean resizeAlignCorners)

Parameters
resizeAlignCorners If true, the centers of the 4 corner pixels of the input and output tensors are aligned, preserving the values at the corner pixels. Defaults to false.