Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/
H A DRNN.cpp670 int mat_numel = c10::multiply_integers(filter_dim_a, filter_dim_a + nb_dims); in add_projection_weights() local
813 int mat_numel = in get_parameters() local