Searched refs:tf_softmax_op (Results 1 – 1 of 1) sorted by relevance
1228 auto tf_softmax_op = cast<TF::SoftmaxOp>(op); in matchAndRewrite() local1231 rewriter, op, tf_softmax_op.getResult(), tf_softmax_op.logits()); in matchAndRewrite()