org.tensorflow.op.xla

Class While

java.lang.Object

org.tensorflow.op.RawOp

org.tensorflow.op.xla.While

All Implemented Interfaces:

Iterable<Operand<TType>>, Op

@Operator(group="xla")
public final class While
extends RawOp
implements Iterable<Operand<TType>>

output = input; While (Cond(output)) { output = Body(output) }

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	While.Inputs

Field Summary

Fields

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

Fields inherited from class org.tensorflow.op.RawOp

operation

Constructor Summary

Constructors

Constructor and Description
While(Operation operation)

Method Summary

Modifier and Type	Method and Description
static While	create(Scope scope, Iterable<Operand<?>> input, ConcreteFunction cond, ConcreteFunction body) Factory method to create a class wrapping a new XlaWhile operation.
Iterator<Operand<TType>>	iterator()
List<Output<?>>	output() Gets output.

Methods inherited from class org.tensorflow.op.RawOp

equals, hashCode, op, toString

Methods inherited from class java.lang.Object

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

Methods inherited from interface java.lang.Iterable

forEach, spliterator

Methods inherited from interface org.tensorflow.op.Op

env

Field Detail

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 Detail

While

public While(Operation operation)

Method Detail

create

@Endpoint(describeByClass=true,
          name="whileOp")
public static While create(Scope scope,
                                                                                     Iterable<Operand<?>> input,
                                                                                     ConcreteFunction cond,
                                                                                     ConcreteFunction body)

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

Parameters:

scope - current scope

input - A list of input tensors whose types are T.

cond - A function takes 'input' and returns a tensor. If the tensor is a scalar of non-boolean, the scalar is converted to a boolean according to the following rule: if the scalar is a numerical value, non-zero means True and zero means False; if the scalar is a string, non-empty means True and empty means False. If the tensor is not a scalar, non-emptiness means True and False otherwise.

body - A function that takes a list of tensors and returns another list of tensors. Both lists have the same types as specified by T.

Returns:

a new instance of While

output

public List<Output<?>> output()

Gets output. A list of output tensors whose types are T.

Returns:

output.

iterator

public Iterator<Operand<TType>> iterator()

Specified by:

iterator in interface Iterable<Operand<TType>>