Searched defs:_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() function
119 _EmbeddingBagKernelCache* /* fbgemm_kernel_cache */) { in index_select_add() argument507 _EmbeddingBagKernelCache* /* fbgemm_kernel_cache */) { in index_select_scale_add() argument