Class InfeedEnqueue

java.lang.Object

org.tensorflow.op.RawOp

org.tensorflow.op.tpu.InfeedEnqueue

All Implemented Interfaces:

Op

@Operator(group="tpu")
public final class InfeedEnqueue
extends RawOp

An op which feeds a single Tensor value into the computation.

Nested Class Summary

Nested Classes

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

Field Summary

Fields

Modifier and Type	Field	Description
static final String	OP_NAME	The name of this op, as known by TensorFlow core engine

Fields inherited from class RawOp

operation

Constructor Summary

Constructors

Constructor	Description
InfeedEnqueue(Operation operation)

Method Summary

Modifier and Type	Method	Description
static InfeedEnqueue	create(Scope scope, Operand<? extends TType> input, InfeedEnqueue.Options... options)	Factory method to create a class wrapping a new InfeedEnqueue operation.
static InfeedEnqueue.Options	deviceOrdinal(Long deviceOrdinal)	Sets the deviceOrdinal option.
static InfeedEnqueue.Options	layout(Long... layout)	Sets the layout option.
static InfeedEnqueue.Options	layout(List<Long> layout)	Sets the layout option.
static InfeedEnqueue.Options	shape(Shape shape)	Sets the shape option.

Methods inherited from class RawOp

equals, hashCode, op, toString

Methods inherited from class Object

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

Methods inherited from interface Op

env

Field Details

OP_NAME

public static final String OP_NAME

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

See Also:

• Constant Field Values

Constructor Details

InfeedEnqueue

public InfeedEnqueue(Operation operation)

Method Details

create

@Endpoint(describeByClass=true)
public static InfeedEnqueue create(Scope scope,
 Operand<? extends TType> input,
 InfeedEnqueue.Options... options)

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

Parameters:

scope - current scope

input - A tensor that will be provided using the infeed mechanism.

options - carries optional attribute values

Returns:

a new instance of InfeedEnqueue

shape

public static InfeedEnqueue.Options shape(Shape shape)

Sets the shape option.

Parameters:

shape - The shape of the tensor.

Returns:

this Options instance.

layout

public static InfeedEnqueue.Options layout(List<Long> layout)

Sets the layout option.

Parameters:

layout - A vector holding the requested layout in minor-to-major sequence. If a layout attribute is passed, but its values are all -1, the layout will be computed by the infeed operation.

Returns:

this Options instance.

layout

public static InfeedEnqueue.Options layout(Long... layout)

Sets the layout option.

Parameters:

layout - A vector holding the requested layout in minor-to-major sequence. If a layout attribute is passed, but its values are all -1, the layout will be computed by the infeed operation.

Returns:

this Options instance.

deviceOrdinal

public static InfeedEnqueue.Options deviceOrdinal(Long deviceOrdinal)

Sets the deviceOrdinal option.

Parameters:

deviceOrdinal - The TPU device to use. This should be -1 when the Op is running on a TPU device, and >= 0 when the Op is running on the CPU device.

Returns:

this Options instance.