Searched refs:tf_stopgrad_op (Results 1 – 1 of 1) sorted by relevance
1927 auto tf_stopgrad_op = cast<TF::StopGradientOp>(op); in matchAndRewrite() local1929 tf_stopgrad_op.getResult().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()1933 tf_stopgrad_op.input()); in matchAndRewrite()