@@ -1187,10 +1187,10 @@ std::tuple<Tensor, Tensor, Tensor, Tensor, Tensor> _thnn_fused_lstm_cell_backwar
1187
1187
DEFINE_DISPATCH (NAME##_miopen_stub); \
1188
1188
DEFINE_DISPATCH (NAME##_packed_cudnn_stub); \
1189
1189
DEFINE_DISPATCH (NAME##_packed_miopen_stub); \
1190
- REGISTER_NO_CPU_DISPATCH (NAME##_cudnn_stub); \
1191
- REGISTER_NO_CPU_DISPATCH (NAME##_miopen_stub); \
1192
- REGISTER_NO_CPU_DISPATCH (NAME##_packed_cudnn_stub); \
1193
- REGISTER_NO_CPU_DISPATCH (NAME##_packed_miopen_stub); \
1190
+ REGISTER_NO_CPU_DISPATCH (NAME##_cudnn_stub) \
1191
+ REGISTER_NO_CPU_DISPATCH(NAME##_miopen_stub) \
1192
+ REGISTER_NO_CPU_DISPATCH(NAME##_packed_cudnn_stub) \
1193
+ REGISTER_NO_CPU_DISPATCH(NAME##_packed_miopen_stub) \
1194
1194
\
1195
1195
std::tuple<Tensor, Tensor> NAME( \
1196
1196
const Tensor& _input, \
@@ -1415,17 +1415,17 @@ static std::tuple<Tensor, Tensor> quantized_gru_data_legacy(
1415
1415
using tanf_cell_type = SimpleCell<tanh_f, CellParams>;
1416
1416
ONE_HIDDEN_RNN (rnn_tanh, tanf_cell_type)
1417
1417
using relu_cell_type = SimpleCell<relu_f, CellParams>;
1418
- ONE_HIDDEN_RNN (rnn_relu, relu_cell_type);
1418
+ ONE_HIDDEN_RNN (rnn_relu, relu_cell_type)
1419
1419
1420
1420
DEFINE_DISPATCH (lstm_cudnn_stub);
1421
1421
DEFINE_DISPATCH (lstm_packed_cudnn_stub);
1422
1422
DEFINE_DISPATCH (lstm_miopen_stub);
1423
1423
DEFINE_DISPATCH (lstm_packed_miopen_stub);
1424
1424
DEFINE_DISPATCH (lstm_mkldnn_stub);
1425
- REGISTER_NO_CPU_DISPATCH (lstm_cudnn_stub);
1426
- REGISTER_NO_CPU_DISPATCH (lstm_packed_cudnn_stub);
1427
- REGISTER_NO_CPU_DISPATCH (lstm_miopen_stub);
1428
- REGISTER_NO_CPU_DISPATCH (lstm_packed_miopen_stub);
1425
+ REGISTER_NO_CPU_DISPATCH (lstm_cudnn_stub)
1426
+ REGISTER_NO_CPU_DISPATCH (lstm_packed_cudnn_stub)
1427
+ REGISTER_NO_CPU_DISPATCH (lstm_miopen_stub)
1428
+ REGISTER_NO_CPU_DISPATCH (lstm_packed_miopen_stub)
1429
1429
1430
1430
std::tuple<Tensor, Tensor, Tensor> lstm (
1431
1431
const Tensor& _input, TensorList hx,
@@ -1857,9 +1857,9 @@ static std::tuple<Tensor, Tensor> prepare_quantized_lstm_hx(TensorList hx) {
1857
1857
// Quantized LSTM cell
1858
1858
using quantized_lstm_cell_dynamic_type = LSTMCell<QuantizedCellParamsDynamic>;
1859
1859
1860
- DEFINE_QUANTIZED_RNN_CELL (quantized_lstm_cell, TensorList, quantized_lstm_cell_type, quantized_lstm_return_type, prepare_quantized_lstm_hx);
1860
+ DEFINE_QUANTIZED_RNN_CELL (quantized_lstm_cell, TensorList, quantized_lstm_cell_type, quantized_lstm_return_type, prepare_quantized_lstm_hx)
1861
1861
1862
- static DEFINE_QUANTIZED_RNN_CELL_DYNAMIC (quantized_lstm_cell_dynamic, TensorList, quantized_lstm_cell_dynamic_type, quantized_lstm_return_type, prepare_quantized_lstm_hx);
1862
+ static DEFINE_QUANTIZED_RNN_CELL_DYNAMIC (quantized_lstm_cell_dynamic, TensorList, quantized_lstm_cell_dynamic_type, quantized_lstm_return_type, prepare_quantized_lstm_hx)
1863
1863
1864
1864
// Helpers for simpler cells
1865
1865
using simple_hx_type = const Tensor&;
@@ -1871,21 +1871,21 @@ static simple_hx_type prepare_quantized_hx(simple_hx_type hx) {
1871
1871
using quantized_gru_cell_type = GRUCell<QuantizedCellParams>;
1872
1872
using quantized_gru_cell_dynamic_type = GRUCell<QuantizedCellParamsDynamic>;
1873
1873
1874
- DEFINE_QUANTIZED_RNN_CELL (quantized_gru_cell, simple_hx_type, quantized_gru_cell_type, Tensor, prepare_quantized_hx);
1874
+ DEFINE_QUANTIZED_RNN_CELL (quantized_gru_cell, simple_hx_type, quantized_gru_cell_type, Tensor, prepare_quantized_hx)
1875
1875
1876
- static DEFINE_QUANTIZED_RNN_CELL_DYNAMIC (quantized_gru_cell_dynamic, simple_hx_type, quantized_gru_cell_dynamic_type, Tensor, prepare_quantized_hx);
1876
+ static DEFINE_QUANTIZED_RNN_CELL_DYNAMIC (quantized_gru_cell_dynamic, simple_hx_type, quantized_gru_cell_dynamic_type, Tensor, prepare_quantized_hx)
1877
1877
1878
1878
// Quantized RNN w/ ReLU cell
1879
1879
using quantized_rnn_relu_cell_type = SimpleCell<relu_f, QuantizedCellParams>;
1880
- DEFINE_QUANTIZED_RNN_CELL (quantized_rnn_relu_cell, simple_hx_type, quantized_rnn_relu_cell_type, Tensor, prepare_quantized_hx);
1880
+ DEFINE_QUANTIZED_RNN_CELL (quantized_rnn_relu_cell, simple_hx_type, quantized_rnn_relu_cell_type, Tensor, prepare_quantized_hx)
1881
1881
using quantized_rnn_relu_cell_dynamic_type = SimpleCell<relu_f, QuantizedCellParamsDynamic>;
1882
- static DEFINE_QUANTIZED_RNN_CELL_DYNAMIC (quantized_rnn_relu_cell_dynamic, simple_hx_type, quantized_rnn_relu_cell_dynamic_type, Tensor, prepare_quantized_hx);
1882
+ static DEFINE_QUANTIZED_RNN_CELL_DYNAMIC (quantized_rnn_relu_cell_dynamic, simple_hx_type, quantized_rnn_relu_cell_dynamic_type, Tensor, prepare_quantized_hx)
1883
1883
1884
1884
// Quantized RNN w/ tanh cell
1885
1885
using quantized_rnn_tanh_cell_type = SimpleCell<tanh_f, QuantizedCellParams>;
1886
- DEFINE_QUANTIZED_RNN_CELL (quantized_rnn_tanh_cell, simple_hx_type, quantized_rnn_tanh_cell_type, Tensor, prepare_quantized_hx);
1886
+ DEFINE_QUANTIZED_RNN_CELL (quantized_rnn_tanh_cell, simple_hx_type, quantized_rnn_tanh_cell_type, Tensor, prepare_quantized_hx)
1887
1887
using quantized_rnn_tanh_cell_dynamic_type = SimpleCell<tanh_f, QuantizedCellParamsDynamic>;
1888
- static DEFINE_QUANTIZED_RNN_CELL_DYNAMIC (quantized_rnn_tanh_cell_dynamic, simple_hx_type, quantized_rnn_tanh_cell_dynamic_type, Tensor, prepare_quantized_hx);
1888
+ static DEFINE_QUANTIZED_RNN_CELL_DYNAMIC (quantized_rnn_tanh_cell_dynamic, simple_hx_type, quantized_rnn_tanh_cell_dynamic_type, Tensor, prepare_quantized_hx)
1889
1889
1890
1890
namespace {
1891
1891
0 commit comments