Searched refs:embedding_pruning_mask (Results 1 – 4 of 4) sorted by relevance
48 const Model_::EmbeddingPruningMask* embedding_pruning_mask) { in FromBuffer() argument86 embedding_pruning_mask)); in FromBuffer()94 const Model_::EmbeddingPruningMask* embedding_pruning_mask) in TFLiteEmbeddingExecutor() argument103 if ((embedding_pruning_mask != nullptr) && in TFLiteEmbeddingExecutor()104 (embedding_pruning_mask->enabled())) { in TFLiteEmbeddingExecutor()105 for (int i = 0; i < embedding_pruning_mask->pruning_mask()->size(); i++) { in TFLiteEmbeddingExecutor()106 pruning_mask_.push_back((*(embedding_pruning_mask->pruning_mask()))[i]); in TFLiteEmbeddingExecutor()109 full_num_buckets_ = embedding_pruning_mask->full_num_buckets(); in TFLiteEmbeddingExecutor()110 pruned_row_bucket_id_ = embedding_pruning_mask->pruned_row_bucket_id(); in TFLiteEmbeddingExecutor()
82 const Model_::EmbeddingPruningMask* embedding_pruning_mask = nullptr);102 const Model_::EmbeddingPruningMask* embedding_pruning_mask = nullptr);
453 embedding_pruning_mask:Model_.EmbeddingPruningMask;
356 model_->embedding_pruning_mask()); in ValidateAndInitialize()