tensorflow :: ops :: FusedPadConv2D

#include <nn_ops.h>

Esegue un riempimento come pre-processo durante una convoluzione.

Sommario

Simile a FusedResizeAndPadConv2d, questa opzione 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 impedisce la necessità di scrivere i risultati intermedi come interi tensori, riducendo la pressione della 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 utilizzato l'ordine "NHWC". Internamente questa operazione utilizza un singolo buffer di memoria virtuale 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] .
  • padding: una matrice a due colonne che specifica le dimensioni del padding. Il numero di righe deve essere uguale al rango di input .
  • filtro: 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 il formato.
  • padding: il tipo di algoritmo di riempimento da utilizzare.

Ritorna:

  • Output : il tensore di output.

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

operator :: tensorflow :: Input

 operator::tensorflow::Input() const

operator :: tensorflow :: Output

 operator::tensorflow::Output() const