Searched refs:rewrite_training_op_ (Results 1 – 2 of 2) sorted by relevance
35 : rewrite_training_op_(rewrite_training_op), in rewrite_training_op_() function46 bool rewrite_training_op_;
77 rewrite_training_op_(rewrite_training_op), in BatchNormExpanderVisitor()162 bool rewrite_training_op_; member in xla::__anon262de65b0111::BatchNormExpanderVisitor187 if (!rewrite_training_op_) { in HandleBatchNormTraining()614 if (BatchNormExpanderVisitor::Run(comp, rewrite_training_op_, in Run()