classe estática pública InfeedEnqueueTuple.Options
Atributos opcionais para InfeedEnqueueTuple
Métodos Públicos
InfeedEnqueueTuple.Options | deviceOrdinal (dispositivo longo Ordinal) |
InfeedEnqueueTuple.Options | layouts (lista <longo> 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 estiver sendo executado em um dispositivo TPU e >= 0 quando o Op estiver sendo executado no dispositivo CPU. |
---|
layouts públicos InfeedEnqueueTuple.Options (layouts List<Long>)
Parâmetros
layouts | Um vetor que contém o layout solicitado em sequência menor para maior para todas as formas da 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. |
---|