org.tensorflow.op.tpu

Class PartitionedCall

java.lang.Object

org.tensorflow.op.RawOp

org.tensorflow.op.tpu.PartitionedCall

All Implemented Interfaces:

Iterable<Operand<TType>>, Op

public final class PartitionedCall
extends RawOp
implements Iterable<Operand<TType>>

Calls a function placed on a specified TPU device.

Nested Class Summary

Nested Classes

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

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
PartitionedCall(Operation operation)

Method Summary

Modifier and Type	Method and Description
static PartitionedCall.Options	autotunerThresh(Long autotunerThresh) Sets the autotunerThresh option.
static PartitionedCall	create(Scope scope, Iterable<Operand<?>> args, Operand<TInt32> deviceOrdinal, List<Class<? extends TType>> Tout, ConcreteFunction f, PartitionedCall.Options... options) Factory method to create a class wrapping a new TPUPartitionedCall 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

PartitionedCall

public PartitionedCall(Operation operation)

Method Detail

create

@Endpoint(describeByClass=true)
public static PartitionedCall create(Scope scope,
                                                                     Iterable<Operand<?>> args,
                                                                     Operand<TInt32> deviceOrdinal,
                                                                     List<Class<? extends TType>> Tout,
                                                                     ConcreteFunction f,
                                                                     PartitionedCall.Options... options)

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

Parameters:

scope - current scope

args - The arguments to the function.

deviceOrdinal - The TPU device ordinal to run the function on.

Tout - The types of the outputs of the function.

f - The function to call.

options - carries optional attribute values

Returns:

a new instance of PartitionedCall

autotunerThresh

public static PartitionedCall.Options autotunerThresh(Long autotunerThresh)

Sets the autotunerThresh option.

Parameters:

autotunerThresh - the autotunerThresh option

Returns:

this Options instance.

output

public List<Output<?>> output()

Gets output. The output of the function call.

Returns:

output.

iterator

public Iterator<Operand<TType>> iterator()

Specified by:

iterator in interface Iterable<Operand<TType>>