Searched refs:full_num_buckets (Results 1 – 2 of 2) sorted by relevance
109 full_num_buckets_ = embedding_pruning_mask->full_num_buckets(); in TFLiteEmbeddingExecutor()162 int full_num_buckets; in AddEmbedding() local164 full_num_buckets = full_num_buckets_; in AddEmbedding()166 full_num_buckets = num_buckets_; in AddEmbedding()168 if (bucket_id >= full_num_buckets) { in AddEmbedding()
515 full_num_buckets:int;