Searched refs:tf_shape_op (Results 1 – 1 of 1) sorted by relevance
699 auto tf_shape_op = cast<TF::ShapeOp>(op); in matchAndRewrite() local702 tf_shape_op.getResult().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()707 tf_shape_op.input().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()