Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DRNN.cpp598 …ector<CellParams> gather_params(TensorList params, bool has_biases, bool has_projections = false) { in gather_params()
1450 bool has_projections = (hx[0].sym_size(2) != hx[1].sym_size(2)); in lstm() local
1503 bool has_projections = (hx[0].size(2) != hx[1].size(2)); in lstm() local
/aosp_15_r20/external/pytorch/torch/_decomp/
H A Ddecompositions.py2951 def gather_params(params, has_biases, has_projections): argument