public static class InfeedEnqueueTuple.Options
Attributi facoltativi per InfeedEnqueueTuple
Metodi pubblici
| InfeedEnqueueTuple.Options | deviceOrdinal (dispositivo lungoOrdinale) |
| InfeedEnqueueTuple.Options | layout (List <Long> layouts) |
Metodi ereditati
Metodi pubblici
public InfeedEnqueueTuple.Options deviceOrdinal (dispositivo lungoOrdinal)
Parametri
| deviceOrdinal | Il dispositivo TPU da utilizzare. Dovrebbe essere -1 quando Op è in esecuzione su un dispositivo TPU e> = 0 quando Op è in esecuzione sul dispositivo CPU. |
|---|
layout public InfeedEnqueueTuple.Options ( layout List <Long>)
Parametri
| layout | Un vettore che contiene il layout richiesto in sequenza da minore a maggiore per tutte le forme della tupla, nell'ordine in cui le forme appaiono nell'input "forme". Gli elementi di layout per una sottoforma possono essere impostati su -1, nel qual caso il layout corrispondente verrà calcolato dall'operazione di alimentazione. |
|---|