Searched refs:fbgemm_kernel_cache (Results 1 – 2 of 2) sorted by relevance
200 _EmbeddingBagKernelCache* fbgemm_kernel_cache) { in index_select_add() argument228 auto kernel_16bit_index_t = fbgemm_kernel_cache in index_select_add()229 ? fbgemm_kernel_cache in index_select_add()378 _EmbeddingBagKernelCache* fbgemm_kernel_cache) { in index_select_add() argument407 fbgemm_kernel_cache ? in index_select_add()408 fbgemm_kernel_cache->getCallback</* has_weight */ false, index_t, float>(ddim) : in index_select_add()565 _EmbeddingBagKernelCache* fbgemm_kernel_cache) { in index_select_scale_add() argument607 auto kernel_16bit_index_t = fbgemm_kernel_cache in index_select_scale_add()608 ? fbgemm_kernel_cache in index_select_scale_add()754 _EmbeddingBagKernelCache* fbgemm_kernel_cache) { in index_select_scale_add() argument[all …]
135 _EmbeddingBagKernelCache* fbgemm_kernel_cache = nullptr);151 _EmbeddingBagKernelCache* fbgemm_kernel_cache = nullptr);