Searched refs:tf_broadcast_to_op (Results 1 – 1 of 1) sorted by relevance
854 auto tf_broadcast_to_op = cast<TF::BroadcastToOp>(op); in matchAndRewrite() local855 auto input_type = tf_broadcast_to_op.input().getType().cast<ShapedType>(); in matchAndRewrite()856 auto output_type = tf_broadcast_to_op.output().getType().cast<ShapedType>(); in matchAndRewrite()857 auto shape_type = tf_broadcast_to_op.shape().getType().cast<ShapedType>(); in matchAndRewrite()878 op->getLoc(), output_type, tf_broadcast_to_op.shape(), in matchAndRewrite()882 op->getLoc(), output_type, tf_broadcast_to_op.input(), tf_fill_op); in matchAndRewrite()