Package org.tensorflow.op.nn

Class FusedResizeAndPadConv2d<T extends java.lang.Number>

java.lang.Object

org.tensorflow.op.PrimitiveOp

org.tensorflow.op.nn.FusedResizeAndPadConv2d<T>

Type Parameters:

T - data type for output() output

All Implemented Interfaces:

Op, Operand<T>

@Operator(group="nn")
public final class FusedResizeAndPadConv2d<T extends java.lang.Number>
extends PrimitiveOp
implements Operand<T>

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 Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	FusedResizeAndPadConv2d.Options	Optional attributes for FusedResizeAndPadConv2d

Field Summary

Fields inherited from class org.tensorflow.op.PrimitiveOp

operation

Method Summary

Modifier and Type	Method	Description
Output<T>	asOutput()	Returns the symbolic handle of a tensor.
static <T extends java.lang.Number> FusedResizeAndPadConv2d<T>	create(Scope scope, Operand<T> input, Operand<java.lang.Integer> size, Operand<java.lang.Integer> paddings, Operand<T> filter, java.lang.String mode, java.util.List<java.lang.Long> strides, java.lang.String padding, FusedResizeAndPadConv2d.Options... options)	Factory method to create a class to wrap a new FusedResizeAndPadConv2d operation to the graph.
Output<T>	output()
static FusedResizeAndPadConv2d.Options	resizeAlignCorners(java.lang.Boolean resizeAlignCorners)

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Methods inherited from class org.tensorflow.op.PrimitiveOp

equals, hashCode, toString

Method Detail

create

public static <T extends java.lang.Number> FusedResizeAndPadConv2d<T> create(Scope scope,
                                                                             Operand<T> input,
                                                                             Operand<java.lang.Integer> size,
                                                                             Operand<java.lang.Integer> paddings,
                                                                             Operand<T> filter,
                                                                             java.lang.String mode,
                                                                             java.util.List<java.lang.Long> strides,
                                                                             java.lang.String padding,
                                                                             FusedResizeAndPadConv2d.Options... options)

Factory method to create a class to wrap a new FusedResizeAndPadConv2d operation to the graph.

Parameters:

scope - current graph 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]`.

mode -

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

resizeAlignCorners

public static FusedResizeAndPadConv2d.Options resizeAlignCorners(java.lang.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.

output

public Output<T> output()

asOutput

public Output<T> asOutput()

Description copied from interface: Operand

Returns the symbolic handle of a 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.

Specified by:

asOutput in interface Operand<T extends java.lang.Number>

See Also:

OperationBuilder.addInput(Output)