Don't auto-cluster must-be-constant ops with side effects.
Clustering must-be-constant side effecting ops is a problem because we can't constant fold these even with constant inputs. This CL is organized as follows: - tf2xla/const_analysis can now report not just the arg indices of must-be-constant values, but also their node ids. - jit/mark_for_compilation_pass consults const_analysis and avoids clustering side effecting ops that must be constant. PiperOrigin-RevId: 209996730
Loading
Please sign in to comment