InfeedEnqueueTuple.Options classe estática pública
Atributos opcionais para InfeedEnqueueTuple
Métodos Públicos
InfeedEnqueueTuple.Options | deviceOrdinal (Long deviceOrdinal) |
InfeedEnqueueTuple.Options | layouts (List <longo> layouts) |
Métodos herdados
Métodos Públicos
pública InfeedEnqueueTuple.Options deviceOrdinal (Long deviceOrdinal)
Parâmetros
deviceOrdinal | O dispositivo TPU a ser usado. Deve ser -1 quando o Op está em execução em um dispositivo TPU e> = 0 quando o Op está em execução no dispositivo de CPU. |
---|
públicas InfeedEnqueueTuple.Options layouts (List <longo> layouts)
Parâmetros
layouts | Um vetor que mantém o layout solicitado em sequência de menor para maior para todas as formas de tupla, na ordem em que as formas aparecem na entrada "formas". Os elementos de layout para uma subforma podem ser definidos como -1, caso em que o layout correspondente será calculado pela operação de alimentação. |
---|