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.

Ritorna:

  • 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