Searched defs:packed_input (Results 1 – 4 of 4) sorted by relevance
532 const PackedSequence& packed_input, in forward_with_packed_input()683 const PackedSequence& packed_input, in forward_with_packed_input()779 const PackedSequence& packed_input, in forward_with_packed_input()
724 torch::nn::utils::rnn::PackedSequence packed_input = in TEST_F() local742 torch::nn::utils::rnn::PackedSequence packed_input = in TEST_F() local760 torch::nn::utils::rnn::PackedSequence packed_input = in TEST_F() local
310 const auto& packed_input = packed_input_and_idx.value(); in SetReplicateOpOperands() local458 for (auto packed_input : packed_inputs) { in BuildReplicateOp() local
447 def func(packed_input): argument