Home
last modified time | relevance | path

Searched defs:batch_first (Results 1 – 25 of 34) sorted by relevance

12

/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/
H A DRNN.cpp42 bool batch_first, in mkldnn_rnn_layer()
69 bool batch_first, in mkldnn_rnn_layer_backward()
93 bool batch_first; member
230 bool batch_first, in mkldnn_rnn_layer()
316 bool batch_first, in mkldnn_rnn_layer_backward()
458 int64_t num_layers, bool has_biases, bool batch_first, double dropout_p, in mkldnn_rnn()
538 int64_t num_layers, double dropout_p, bool train, bool bidirectional, bool batch_first) { in mkldnn_impl()
553 int64_t num_layers, double dropout_p, bool train, bool bidirectional, bool batch_first) { in lstm_mkldnn()
H A DMKLDNNConversions.cpp468 bool batch_first, in mkldnn_reorder_mkldnn_rnn_layer_weight()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DPackedSequence.cpp34 …or, Tensor> _pack_padded_sequence(const Tensor& _input, const Tensor& _lengths, bool batch_first) { in _pack_padded_sequence()
115 …const Tensor& grad, c10::SymIntArrayRef input_size, const Tensor& _batch_sizes, bool batch_first) { in _pack_padded_sequence_backward_symint()
142 …pad_packed_sequence(const Tensor& data, const Tensor& _batch_sizes, bool batch_first, const Scalar… in _pad_packed_sequence()
205 Tensor pad_sequence(TensorList sequences, bool batch_first, double padding_value, const c10::string… in pad_sequence()
H A DRNN.cpp1433 int64_t num_layers, double dropout_p, bool train, bool bidirectional, bool batch_first) { in lstm()
1697 bool batch_first, in quantized_lstm_input()
1737 bool batch_first, in quantized_lstm_input_legacy()
/aosp_15_r20/external/pytorch/test/onnx/model_defs/
H A Drnn_model_with_packed_sequence.py6 def __init__(self, model, batch_first): argument
21 def __init__(self, model, batch_first): argument
35 def __init__(self, model, batch_first): argument
H A Dlstm_flattening_result.py12 def __init__(self, input_size, hidden_size, layers, bidirect, dropout, batch_first): argument
31 def __init__(self, input_size, hidden_size, layers, bidirect, dropout, batch_first): argument
/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/
H A DGru.cpp51 bool batch_first) { in gru_input()
194 bool batch_first) { in GruPackedContext()
269 bool batch_first) { in create_gru_context()
292 const bool batch_first = in run_gru_context() local
H A DLstm.cpp60 bool batch_first) { in lstm_input()
236 bool batch_first) { in LstmPackedContext()
311 bool batch_first) { in create_lstm_context()
338 const bool batch_first = in run_lstm_context() local
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dnn_utils.cpp475 for (bool batch_first : std::vector<bool>{true, false}) { in TEST_F() local
490 for (bool batch_first : std::vector<bool>{true, false}) { in TEST_F() local
553 bool enforce_sorted = false) { in TEST_F()
637 for (bool batch_first : std::vector<bool>{true, false}) { in TEST_F() local
725 for (bool batch_first : std::vector<bool>{true, false}) { in TEST_F() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/
H A DRNN_miopen.cpp34 bool batch_first, double fn_dropout, bool fn_train, bool fn_bidirectional, in miopen_rnn()
42 …al<Tensor>& grad_cy_r_opt, int64_t mode, int64_t hidden_size, int64_t num_layers, bool batch_first, in miopen_rnn_backward()
449 bool batch_first, double fn_dropout, bool fn_train, bool fn_bidirectional, in miopen_rnn()
562 int64_t fn_num_layers, bool batch_first, double fn_dropout, in miopen_rnn_backward_input()
677 int64_t fn_num_layers, bool batch_first, double fn_dropout, in miopen_rnn_backward_weight()
762 …al<Tensor>& grad_cy_r_opt, int64_t mode, int64_t hidden_size, int64_t num_layers, bool batch_first, in miopen_rnn_backward()
847 int64_t num_layers, double dropout_p, bool train, bool bidirectional, bool batch_first) { in _miopen_impl()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/
H A DRNN.cpp45 bool batch_first, in _cudnn_rnn_flatten_weight()
61 bool batch_first, in _cudnn_rnn()
85 bool batch_first, in _cudnn_rnn_backward()
1247 bool batch_first, in copy_weights_to_flat_buf_views()
1356 bool batch_first, in _cudnn_rnn_flatten_weight()
1392 bool batch_first, in _cudnn_rnn()
1682 bool batch_first, in _cudnn_rnn_backward_input()
1912 bool batch_first, in _cudnn_rnn_backward_weight()
2103 bool batch_first, in _cudnn_rnn_backward()
2582 bool batch_first) { in _cudnn_impl()
/aosp_15_r20/external/pytorch/benchmarks/fastrnns/
H A Dcustom_lstms.py40 batch_first=False, argument
76 batch_first=False, argument
/aosp_15_r20/external/pytorch/torch/nn/utils/
H A D_per_sample_grad.py16 batch_first=True, argument
/aosp_15_r20/external/pytorch/aten/src/ATen/miopen/
H A DAutocastRNN.cpp21 bool batch_first, in miopen_rnn()
/aosp_15_r20/external/pytorch/aten/src/ATen/cudnn/
H A DAutocastRNN.cpp32 bool batch_first, in _cudnn_rnn_cast_reflatten()
/aosp_15_r20/external/pytorch/test/nn/
H A Dtest_packed_sequence.py293 def _compatibility_test(sequences, lengths, batch_first, enforce_sorted=False): argument
/aosp_15_r20/external/pytorch/test/
H A Dtest_transformers.py458 def test_transformerencoder(self, batch_first, training, enable_nested_tensor, device): argument
459 def get_a_test_layer(activation, batch_first=False): argument
488 def _test(batch_first, training, enable_nested_tensor): argument
H A Dtest_expanded_weights.py619 batch_first=True, argument
/aosp_15_r20/external/pytorch/torch/_decomp/
H A Ddecompositions.py2635 def pad_sequence(sequences, batch_first=False, padding_value=0.0): argument
3140 batch_first, argument
3184 batch_first, argument
3215 batch_first, argument
3470 batch_first, argument
3585 batch_first, argument
/aosp_15_r20/external/pytorch/torch/onnx/
H A Dsymbolic_opset9.py4276 batch_first=None, argument
4502 batch_first, argument
4613 batch_first, argument
4696 def _pack_padded_sequence(g: jit_utils.GraphContext, input, lengths, batch_first): argument
4725 batch_first, argument
/aosp_15_r20/external/pytorch/torch/_subclasses/
H A Dfake_impls.py722 def _pack_padded_sequence(fake_mode, func, inputs, lengths, batch_first): argument
/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dvulkan_api_test.cpp7419 const bool batch_first = true; in TEST_F() local
7488 const bool batch_first = true; in TEST_F() local
7553 const bool batch_first = true; in TEST_F() local
7648 const bool batch_first = true; in TEST_F() local
7719 const bool batch_first = true; in TEST_F() local
7926 const bool batch_first = true; in TEST_F() local
8006 const bool batch_first = true; in TEST_F() local
8082 const bool batch_first = true; in TEST_F() local
/aosp_15_r20/external/pytorch/torch/nn/modules/
H A Dactivation.py1050 batch_first=False, argument
/aosp_15_r20/external/pytorch/torch/ao/nn/quantized/dynamic/modules/
H A Drnn.py103 batch_first=False, argument
/aosp_15_r20/external/pytorch/test/quantization/eager/
H A Dtest_quantize_eager_ptq.py1074 def __init__(self, d_model, nhead, batch_first): argument

12