Tune in to the first Women in ML Symposium this Tuesday, October 19 at 9am PST Register now

XlaSendToHost

public final class XlaSendToHost

An op to send a tensor to the host.

input: the tensor that will be sent to the host. Tinput: element type for input. key: A unique identifier for this region used to match up host transfers.

Public Methods

static <T> XlaSendToHost
create(Scope scope, Operand<T> input, String key)
Factory method to create a class wrapping a new XlaSendToHost operation.

Inherited Methods

Public Methods

public static XlaSendToHost create (Scope scope, Operand<T> input, String key)

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

Parameters
scope current scope
Returns
  • a new instance of XlaSendToHost