Searched refs:tf_reshape_op (Results 1 – 1 of 1) sorted by relevance
656 auto tf_reshape_op = cast<TF::ReshapeOp>(op); in matchAndRewrite() local659 tf_reshape_op.getResult().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()672 op, output_type, tf_reshape_op.tensor(), shape_attr); in matchAndRewrite()