Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dregister_prim_ops_fulljit.cpp390 std::vector<int64_t> _output_size( in _output_size() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/
H A DRNN.cpp172 std::vector<int64_t> _output_size(const RNNParams& rnn) { in _output_size() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/
H A DRNN_miopen.cpp437 std::vector<int64_t> _output_size(const RNNDescriptorParams& rnn, const TensorDescriptorListParams&… in _output_size() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/
H A DRNN.cpp1094 std::vector<int64_t> _output_size( in get_parameters() function