Searched defs:compressed_indices_mapping_data (Results 1 – 2 of 2) sorted by relevance
51 int32_t* compressed_indices_mapping_data = nullptr; in embedding_lookup_fallback_impl() local240 int32_t* compressed_indices_mapping_data = nullptr; in embedding_bag_nbit_impl() local412 int32_t* compressed_indices_mapping_data = nullptr; in embedding_bag_byte_impl() local
40 auto compressed_indices_mapping_data = in _rowwise_prune_helper() local