Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/
H A DRNN_miopen.cpp425 std::vector<int64_t> _input_size(const TensorDescriptorListParams& tensors) { in _input_size() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/
H A DRNN.cpp1061 std::vector<int64_t> _input_size(const TensorDescriptorListParams& tensors) { in get_parameters() function