tensoreflusso:: ops:: FusedPadConv2D
#include <nn_ops.h>
Esegue un riempimento come preelaborazione durante una convoluzione.
Riepilogo
Simile a FusedResizeAndPadConv2d, questa operazione consente un'implementazione ottimizzata in cui la fase di trasformazione del riempimento spaziale è fusa con la ricerca im2col, ma in questo caso senza il filtro bilineare richiesto per il ridimensionamento. La fusione del riempimento evita la necessità di scrivere i risultati intermedi come interi tensori, riducendo la pressione sulla memoria e possiamo ottenere alcuni guadagni di latenza unendo i calcoli di trasformazione. L'attributo data_format per Conv2D non è supportato da questa operazione e viene invece utilizzato l'ordine "NHWC". Internamente questa operazione utilizza un singolo buffer di scratch per grafico, il che significa che si bloccherà se più versioni vengono eseguite in parallelo. Questo perché questo operatore è principalmente un'ottimizzazione per ridurre al minimo l'utilizzo della memoria.
Argomenti:
- scope: un oggetto Scope
- input: 4-D con forma
[batch, in_height, in_width, in_channels]
. - riempimenti: una matrice a due colonne che specifica le dimensioni del riempimento. Il numero di righe deve essere uguale al rango
input
. - filter: 4-D con forma
[filter_height, filter_width, in_channels, out_channels]
. - passi: 1-D di lunghezza 4. Il passo della finestra scorrevole per ogni dimensione di
input
. Deve essere nello stesso ordine della dimensione specificata con format. - riempimento: il tipo di algoritmo di riempimento da utilizzare.
Resi:
-
Output
: il tensore di uscita.
Costruttori e distruttori | |
---|---|
FusedPadConv2D (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input paddings, :: tensorflow::Input filter, StringPiece mode, const gtl::ArraySlice< int > & strides, StringPiece padding) |
Attributi pubblici | |
---|---|
operation | |
output |
Funzioni pubbliche | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Attributi pubblici
operazione
Operation operation
produzione
::tensorflow::Output output
Funzioni pubbliche
FusedPadConv2D
FusedPadConv2D( const ::tensorflow::Scope & scope, ::tensorflow::Input input, ::tensorflow::Input paddings, ::tensorflow::Input filter, StringPiece mode, const gtl::ArraySlice< int > & strides, StringPiece padding )
nodo
::tensorflow::Node * node() const
operatore::tensorflow::Input
operator::tensorflow::Input() const
operatore::tensorflow::Output
operator::tensorflow::Output() const