przepływ tensorowy:: ops:: FusedPadConv2D

#include <nn_ops.h>

Wykonuje dopełnienie jako proces wstępny podczas splotu.

Streszczenie

Podobnie jak w przypadku FusedResizeAndPadConv2d, ta opcja pozwala na zoptymalizowaną implementację, w której etap transformacji przestrzennego dopełnienia jest połączony z wyszukiwaniem im2col, ale w tym przypadku bez filtrowania dwuliniowego wymaganego do zmiany rozmiaru. Połączenie dopełnienia zapobiega konieczności zapisywania wyników pośrednich jako całych tensorów, co zmniejsza obciążenie pamięci, a dzięki łączeniu obliczeń transformacji możemy uzyskać pewne zyski w zakresie opóźnień. Atrybut data_format dla Conv2D nie jest obsługiwany przez tę operację i zamiast tego używana jest kolejność „NHWC”. Wewnętrznie ta operacja wykorzystuje pojedynczy bufor magazynujący dla każdego wykresu, co oznacza, że ​​będzie blokować, jeśli wiele wersji będzie uruchamianych równolegle. Dzieje się tak, ponieważ ten operator jest przede wszystkim optymalizacją mającą na celu zminimalizowanie zużycia pamięci.

Argumenty:

  • zakres: Obiekt Scope
  • wejście: 4-D z kształtem [batch, in_height, in_width, in_channels] .
  • dopełnienia: dwukolumnowa macierz określająca rozmiary dopełnienia. Liczba wierszy musi być taka sama jak ranga input .
  • filtr: 4-D z kształtem [filter_height, filter_width, in_channels, out_channels] .
  • kroki: 1-D o długości 4. Krok przesuwanego okna dla każdego wymiaru input . Musi być w tej samej kolejności, co wymiar określony w formacie.
  • dopełnienie: typ algorytmu dopełniania, który ma zostać użyty.

Zwroty:

Konstruktory i destruktory

FusedPadConv2D (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input paddings, :: tensorflow::Input filter, StringPiece mode, const gtl::ArraySlice< int > & strides, StringPiece padding)

Atrybuty publiczne

operation
output

Funkcje publiczne

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Atrybuty publiczne

operacja

Operation operation

wyjście

::tensorflow::Output output

Funkcje publiczne

FusedPadConv2D

 FusedPadConv2D(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input paddings,
  ::tensorflow::Input filter,
  StringPiece mode,
  const gtl::ArraySlice< int > & strides,
  StringPiece padding
)

węzeł

::tensorflow::Node * node() const 

operator::tensorflow::Wejście

 operator::tensorflow::Input() const 

operator::tensorflow::Wyjście

 operator::tensorflow::Output() const