Searched refs:training_op (Results 1 – 2 of 2) sorted by relevance
385 training_op = self._mini_batch_training_op(389 training_op = self._full_batch_training_op(393 init_op, training_op)
1843 auto training_op = rewriter.create<BatchNormGradOp>( in matchAndRewrite() local1848 rewriter.create<GetTupleElementOp>(loc, training_op.getResult(), 0); in matchAndRewrite()1851 rewriter.create<GetTupleElementOp>(loc, training_op.getResult(), 1); in matchAndRewrite()1854 rewriter.create<GetTupleElementOp>(loc, training_op.getResult(), 2); in matchAndRewrite()