Searched refs:tf_argmax_op (Results 1 – 1 of 1) sorted by relevance
464 auto tf_argmax_op = cast<TF::ArgMaxOp>(op); in matchAndRewrite() local467 tf_argmax_op.input().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()469 tf_argmax_op.getResult().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()474 if (!matchPattern(tf_argmax_op.dimension(), m_Constant(&axis_elems))) in matchAndRewrite()489 tf_argmax_op.input(), axis_attr); in matchAndRewrite()