TPUCopyWithDynamicShape

public final class TPUCopyWithDynamicShape

Op that copies host tensor to device with dynamic shape support. For internal use only.

Public Methods

static TPUCopyWithDynamicShape
create(Scope scope, Iterable<Operand<?>> tensors, Iterable<Operand<Integer>> unpaddedSizes)
Factory method to create a class wrapping a new TPUCopyWithDynamicShape operation.
Iterator<Operand<Object>>
List<Output<?>>

Inherited Methods

Public Methods

public static TPUCopyWithDynamicShape create (Scope scope, Iterable<Operand<?>> tensors, Iterable<Operand<Integer>> unpaddedSizes)

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

Parameters
scope current scope
Returns
  • a new instance of TPUCopyWithDynamicShape

public Iterator<Operand<Object>> iterator ()

public List<Output<?>> tpuTensors ()