classe estática pública InfeedEnqueueTuple.Options
Atributos opcionais para InfeedEnqueueTuple
Métodos públicos
InfeedEnqueueTuple.Options | deviceOrdinal (Longo deviceOrdinal) |
InfeedEnqueueTuple.Options | layouts (List<Long> layouts) |
Métodos Herdados
Métodos públicos
public InfeedEnqueueTuple.Options deviceOrdinal (long deviceOrdinal)
Parâmetros
dispositivoOrdinal | O dispositivo TPU a ser usado. Deve ser -1 quando o Op está sendo executado em um dispositivo TPU e >= 0 quando o Op está sendo executado no dispositivo CPU. |
---|
Layouts públicos InfeedEnqueueTuple.Options ( layouts List<Long>)
Parâmetros
layouts | Um vetor que mantém o layout solicitado na sequência menor a 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. |
---|