Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/cpp/api/
H A 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
/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_nn.py11291 def check_lengths(lengths, enforce_sorted, use_default_hiddens, proj_size): argument
H A Dtest_jit.py9713 def pack_sequence_func(tensor_list, enforce_sorted=True): argument