Home
last modified time | relevance | path

Searched defs:is_input_packed (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/
H A DRNN_miopen.cpp166 bool is_input_packed() const { in is_input_packed() function
191 auto is_input_packed = batch_sizes.size() != 0; in descriptors() local
470 auto is_input_packed = fn.tensors.batch_sizes.size() != 0; in miopen_rnn() local
582 auto is_input_packed = fn_batch_sizes.size() != 0; in miopen_rnn_backward_input() local
698 auto is_input_packed = fn_batch_sizes.size() != 0; in miopen_rnn_backward_weight() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/
H A DRNN.cpp435 bool is_input_packed() const { in is_input_packed() function
468 auto is_input_packed = batch_sizes.size() != 0; in descriptors() local
477 auto is_input_packed = batch_sizes.size() != 0; in descriptors() local
1452 auto is_input_packed = fn.tensors.batch_sizes.size() != 0; in _cudnn_rnn() local
1729 auto is_input_packed = fn_batch_sizes.size() != 0; in _cudnn_rnn_backward_input() local
1957 auto is_input_packed = fn_batch_sizes.size() != 0; in _cudnn_rnn_backward_weight() local