1#pragma once
2
3// @generated by torchgen/gen.py from NativeFunction.h
4
5#include <c10/core/Scalar.h>
6#include <c10/core/Storage.h>
7#include <c10/core/TensorOptions.h>
8#include <c10/util/Deprecated.h>
9#include <c10/util/Optional.h>
10#include <c10/core/QScheme.h>
11#include <ATen/core/Reduction.h>
12#include <ATen/core/Tensor.h>
13#include <tuple>
14#include <vector>
15
16
17namespace at {
18namespace native {
19TORCH_API at::Tensor & _cudnn_rnn_flatten_weight_out_symint(at::TensorList weight_arr, int64_t weight_stride0, c10::SymInt input_size, int64_t mode, c10::SymInt hidden_size, c10::SymInt proj_size, int64_t num_layers, bool batch_first, bool bidirectional, at::Tensor & out);
20TORCH_API at::Tensor _cudnn_rnn_flatten_weight(at::TensorList weight_arr, int64_t weight_stride0, int64_t input_size, int64_t mode, int64_t hidden_size, int64_t proj_size, int64_t num_layers, bool batch_first, bool bidirectional);
21} // namespace native
22} // namespace at
23