InfeedEnqueueTuple.Options

InfeedEnqueueTuple.Options estáticos de clase pública

Los atributos opcionales para InfeedEnqueueTuple

Métodos públicos

InfeedEnqueueTuple.Options
deviceOrdinal (Long deviceOrdinal)
InfeedEnqueueTuple.Options
diseños (List <Larga> diseños)

Métodos heredados

Métodos públicos

pública InfeedEnqueueTuple.Options deviceOrdinal (Long deviceOrdinal)

Parámetros
deviceOrdinal El dispositivo de TPU que se utilizará. Debe ser -1 cuando el Op se está ejecutando en un dispositivo de TPU y> = 0 cuando el Op se está ejecutando en el dispositivo de la CPU.

públicas InfeedEnqueueTuple.Options diseños (List <Larga> diseños)

Parámetros
diseños Un vector que contiene el diseño solicitado en secuencia de menor a mayor para todas las formas de tupla, en el orden en que aparecen las formas en la entrada "formas". Los elementos de diseño para una subforma se pueden establecer en -1, en cuyo caso la operación de alimentación calculará el diseño correspondiente.