Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DRNN.cpp866 struct FullBidirectionalLayer struct
868 using hidden_type = pair_of<dir_hidden_type>;
869 using param_type = pair_of<cell_params>;
870 using output_type = typename Layer<Tensor, hidden_type, param_type>::output_type;
872 FullBidirectionalLayer(Cell<dir_hidden_type, cell_params>& cell) in FullBidirectionalLayer() function
875 output_type operator()( in operator ()()
911 std::vector<Tensor> reverse(std::vector<Tensor>&& x) const { in reverse()
916 FullLayer<dir_hidden_type, cell_params> layer_;