Searched refs:SimplifyEmbeddingLookupStage (Results 1 – 1 of 1) sorted by relevance
3523 class SimplifyEmbeddingLookupStage : public ArithmeticOptimizerStage { class3525 explicit SimplifyEmbeddingLookupStage( in SimplifyEmbeddingLookupStage() function in tensorflow::grappler::__anon327bfa1e0111::SimplifyEmbeddingLookupStage3530 ~SimplifyEmbeddingLookupStage() override = default;3733 pipeline.AddStage<SimplifyEmbeddingLookupStage>(ctx, ctx_ext); in SimplifyArithmeticOps()