Home
last modified time | relevance | path

Searched refs:matrix_pointer (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/
H A DRNN.cpp600 void* matrix_pointer = nullptr; in add_projection_weights() local
615 /*linLayerMat=*/&matrix_pointer)); in add_projection_weights()
626 /*linLayerMat=*/&matrix_pointer, in add_projection_weights()
661 auto offset_bytes = (char*)matrix_pointer - (char*)weight_buf.data_ptr(); in add_projection_weights()
722 void* matrix_pointer; in get_parameters() local
734 &matrix_pointer)); in get_parameters()
747 &matrix_pointer, in get_parameters()
761 &matrix_pointer)); in get_parameters()
800 (char*)matrix_pointer - (char*)weight_buf.data_ptr(); in get_parameters()
907 void* matrix_pointer; in get_parameters() local
[all …]