ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

XlaRecvFromHost

public final class XlaRecvFromHost

An op to receive a tensor from the host.

output: the tensor that will be received from the host. Toutput: element type for output. shape: shape for output. key: A unique identifier for this region used to match up host transfers.

Public Methods

Output<T>
asOutput()
Returns the symbolic handle of a tensor.
static <T> XlaRecvFromHost<T>
create(Scope scope, Class<T> Toutput, Shape shape, String key)
Factory method to create a class wrapping a new XlaRecvFromHost operation.
Output<T>
output()

Inherited Methods

Public Methods

public Output<T> 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 XlaRecvFromHost<T> create (Scope scope, Class<T> Toutput, Shape shape, String key)

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

Parameters
scope current scope
Returns
  • a new instance of XlaRecvFromHost

public Output<T> output ()