Searched refs:RemoveCastIntoSegmentReductionStage (Results 1 – 1 of 1) sorted by relevance
3609 class RemoveCastIntoSegmentReductionStage : public ArithmeticOptimizerStage { class3611 explicit RemoveCastIntoSegmentReductionStage( in RemoveCastIntoSegmentReductionStage() function in tensorflow::grappler::__anon327bfa1e0111::RemoveCastIntoSegmentReductionStage3616 ~RemoveCastIntoSegmentReductionStage() override = default;3735 pipeline.AddStage<RemoveCastIntoSegmentReductionStage>(ctx, ctx_ext); in SimplifyArithmeticOps()