Searched refs:compressed_indices_mapping_data (Results 1 – 2 of 2) sorted by relevance
40 auto compressed_indices_mapping_data = in _rowwise_prune_helper() local49 compressed_indices_mapping_data[i] = last_row_kept; in _rowwise_prune_helper()52 compressed_indices_mapping_data[i] = -1; in _rowwise_prune_helper()
51 int32_t* compressed_indices_mapping_data = nullptr; in embedding_lookup_fallback_impl() local90 compressed_indices_mapping_data = in embedding_lookup_fallback_impl()95 idx = compressed_indices_mapping_data[uncompressed_idx]; in embedding_lookup_fallback_impl()240 int32_t* compressed_indices_mapping_data = nullptr; in embedding_bag_nbit_impl() local245 compressed_indices_mapping_data = in embedding_bag_nbit_impl()251 compressed_indices_mapping_data[0] == 0)) { in embedding_bag_nbit_impl()355 /*compressed_indices_table=*/compressed_indices_mapping_data); in embedding_bag_nbit_impl()412 int32_t* compressed_indices_mapping_data = nullptr; in embedding_bag_byte_impl() local417 compressed_indices_mapping_data = in embedding_bag_byte_impl()423 compressed_indices_mapping_data[0] == 0)) { in embedding_bag_byte_impl()[all …]