Searched refs:tf_fill_op (Results 1 – 2 of 2) sorted by relevance
768 auto tf_fill_op = cast<TF::FillOp>(op); in matchAndRewrite() local771 tf_fill_op.getResult().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()776 if (!matchPattern(tf_fill_op.dims(), m_Constant(&dims_elems))) in matchAndRewrite()786 if (!matchPattern(tf_fill_op.value(), m_Constant(&value_elem))) in matchAndRewrite()
877 auto tf_fill_op = rewriter.create<TF::FillOp>( in matchAndRewrite() local882 op->getLoc(), output_type, tf_broadcast_to_op.input(), tf_fill_op); in matchAndRewrite()