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