Searched refs:RemoveStackSliceSameAxis (Results 1 – 1 of 1) sorted by relevance
3145 class RemoveStackSliceSameAxis : public ArithmeticOptimizerStage { class3147 explicit RemoveStackSliceSameAxis(const GraphOptimizerContext& ctx, in RemoveStackSliceSameAxis() function in tensorflow::grappler::__anon327bfa1e0111::RemoveStackSliceSameAxis3151 ~RemoveStackSliceSameAxis() override = default;3731 pipeline.AddStage<RemoveStackSliceSameAxis>(ctx, ctx_ext); in SimplifyArithmeticOps()