Home
last modified time | relevance | path

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

/external/pytorch/test/nn/
Dtest_packed_sequence.py293 def _compatibility_test(sequences, lengths, batch_first, enforce_sorted=False): argument
/external/pytorch/test/cpp/api/
Dnn_utils.cpp522 for (bool enforce_sorted : std::vector<bool>{true, false}) { in TEST_F() local
553 bool enforce_sorted = false) { in TEST_F()
638 for (bool enforce_sorted : std::vector<bool>{true, false}) { in TEST_F() local
/external/pytorch/test/
Dtest_nn.py11291 def check_lengths(lengths, enforce_sorted, use_default_hiddens, proj_size): argument
Dtest_jit.py9713 def pack_sequence_func(tensor_list, enforce_sorted=True): argument