InfeedEnqueueTuple.Options

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.