Searched refs:rewrite_grad_op (Results 1 – 2 of 2) sorted by relevance
34 bool rewrite_grad_op = false)37 rewrite_grad_op_(rewrite_grad_op) {} in rewrite_training_op_()
64 bool rewrite_inference_op, bool rewrite_grad_op);75 bool rewrite_grad_op) in BatchNormExpanderVisitor() argument79 rewrite_grad_op_(rewrite_grad_op) {} in BatchNormExpanderVisitor()175 bool rewrite_grad_op) { in Run() argument180 /*rewrite_grad_op=*/rewrite_grad_op); in Run()