Searched refs:rewrite_inference_op_ (Results 1 – 2 of 2) sorted by relevance
36 rewrite_inference_op_(rewrite_inference_op), in rewrite_training_op_()47 bool rewrite_inference_op_; variable
70 rewrite_inference_op_(rewrite_inference_op), in BatchNormExpanderVisitor()133 bool rewrite_inference_op_; member in xla::__anon62d847220111::BatchNormExpanderVisitor295 if (!rewrite_inference_op_) { in HandleBatchNormInference()564 rewrite_inference_op_, in Run()