Home
last modified time | relevance | path

Searched defs:FullLayer (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/aten/src/ATen/native/
DRNN.cpp822 struct FullLayer : Layer<Tensor, hidden_type, cell_params> { struct
823 using output_type =
825 using unstacked_output_type = LayerOutput<std::vector<Tensor>, hidden_type>;
827 FullLayer(Cell<hidden_type, cell_params>& cell) in FullLayer() function
830 unstacked_output_type operator()( in operator ()()
844 output_type operator()( in operator ()()
862 Cell<hidden_type, cell_params>& cell_;