Searched refs:pruned_row_bucket_id_ (Results 1 – 2 of 2) sorted by relevance
110 pruned_row_bucket_id_ = embedding_pruning_mask->pruned_row_bucket_id(); in TFLiteEmbeddingExecutor()141 return pruned_row_bucket_id_; in PruneBucketId()
122 int pruned_row_bucket_id_ = -1; variable