xla

9410cd09 - Apply Optimization Barrier to grad input and saved input (#3721)

Apply Optimization Barrier to grad input and saved input (#3721) * Apply Optimization Barrier to grad input and saved input * move helper function within the class