Searched refs:embedding_lookup_fallback_impl (Results 1 – 1 of 1) sorted by relevance
37 at::Tensor& embedding_lookup_fallback_impl( in embedding_lookup_fallback_impl() function368 return embedding_lookup_fallback_impl<IndexType, OffsetType, 4, 2>( in embedding_bag_nbit_impl()381 return embedding_lookup_fallback_impl<IndexType, OffsetType, 2, 4>( in embedding_bag_nbit_impl()543 return embedding_lookup_fallback_impl<IndexType, OffsetType, 8, 1>( in embedding_bag_byte_impl()