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