Searched refs:_EmbeddingBagKernelCache (Results 1 – 2 of 2) sorted by relevance
113 using _EmbeddingBagKernelCache = _EmbeddingBagKernelCacheImpl< variable123 struct _EmbeddingBagKernelCache { struct124 explicit _EmbeddingBagKernelCache(std::optional<int64_t> /* maybe_block_size */) {} in _EmbeddingBagKernelCache() function135 _EmbeddingBagKernelCache* fbgemm_kernel_cache = nullptr);151 _EmbeddingBagKernelCache* fbgemm_kernel_cache = nullptr);
119 _EmbeddingBagKernelCache* /* fbgemm_kernel_cache */) { in index_select_add() argument200 _EmbeddingBagKernelCache* fbgemm_kernel_cache) { in index_select_add()378 _EmbeddingBagKernelCache* fbgemm_kernel_cache) { in index_select_add()507 _EmbeddingBagKernelCache* /* fbgemm_kernel_cache */) { in index_select_scale_add() argument565 _EmbeddingBagKernelCache* fbgemm_kernel_cache) { in index_select_scale_add()754 _EmbeddingBagKernelCache* fbgemm_kernel_cache) { in index_select_scale_add()1133 … bool include_last_offset, int64_t padding_idx, _EmbeddingBagKernelCache* fbgemm_kernel_cache) { in _embedding_bag_cpu_impl_out()1328 _EmbeddingBagKernelCache* fbgemm_kernel_cache) { in _embedding_bag_cpu_out()