Searched refs:tf_gather_op (Results 1 – 1 of 1) sorted by relevance
1723 auto tf_gather_op = cast<TF::GatherOp>(op); in matchAndRewrite() local1726 tf_gather_op.getResult().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()1734 op, output_type, tf_gather_op.params(), tf_gather_op.indices(), in matchAndRewrite()1742 auto tf_gather_op = cast<TF::GatherV2Op>(op); in matchAndRewrite() local1745 tf_gather_op.getResult().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()1750 if (!matchPattern(tf_gather_op.axis(), m_Constant(&axis_elem))) in matchAndRewrite()1756 auto tmpAttr = tf_gather_op.batch_dimsAttr(); in matchAndRewrite()1764 op, output_type, tf_gather_op.params(), tf_gather_op.indices(), in matchAndRewrite()