Searched refs:pruning_mask_ (Results 1 – 2 of 2) sorted by relevance
106 pruning_mask_.push_back((*(embedding_pruning_mask->pruning_mask()))[i]); in TFLiteEmbeddingExecutor()123 for (const uint64 mask : pruning_mask_) { in ComputePrefixCounts()140 if (!(pruning_mask_[bucket_id_major] & (one << bucket_id_minor))) in PruneBucketId()149 __builtin_popcountll(pruning_mask_[bucket_id_major] & minor_mask); in PruneBucketId()163 if (!pruning_mask_.empty()) { in AddEmbedding()172 if (!pruning_mask_.empty()) { in AddEmbedding()
117 std::vector<uint64> pruning_mask_; variable