Warning: This API is deprecated and will be removed in a future version of TensorFlow after the replacement is stable.

TpuHandleToProtoKey

Stay organized with collections Save and categorize content based on your preferences.
public final class TpuHandleToProtoKey

Converts XRT's uid handles to TensorFlow-friendly input format.

Converts a uid handle for a compiled program into a vector of proto keys.

XRT compile ops return uids, and the TensorFlow execute op takes a proto key. This op enables a client to compile on TPU using XRT and execute using the standard TensorFlow execute op.

'uid' is the input handle. 'proto_keys' is a vector of proto keys, one for each core program.

Public Methods

Output<String>
asOutput()
Returns the symbolic handle of a tensor.
static TpuHandleToProtoKey
create(Scope scope, Operand<Long> uid)
Factory method to create a class wrapping a new TpuHandleToProtoKey operation.
Output<String>

Inherited Methods

Public Methods

public Output<String> asOutput ()

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.

public static TpuHandleToProtoKey create (Scope scope, Operand<Long> uid)

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

Parameters
scope current scope
Returns
  • a new instance of TpuHandleToProtoKey

public Output<String> protoKeys ()