We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see ourdocumentation.
There was an error while loading.Please reload this page.
2 parentsa7331b8 +609e5ad commitc78616fCopy full SHA for c78616f
tensorflow/compiler/mlir/xla/transforms/legalize_tf.cc
@@ -5819,11 +5819,11 @@ class ConvertRandomShuffleOp : public OpRewritePattern<TF::RandomShuffleOp> {
5819
LogicalResultmatchAndRewrite(TF::RandomShuffleOp op,
5820
PatternRewriter &rewriter)constoverride {
5821
auto no_op = [&]() {
5822
- rewriter.replaceOp(op, op.getValue());
+ rewriter.replaceOp(op, op.value());
5823
returnsuccess();
5824
};
5825
5826
-auto input_type = op.getValue().getType().dyn_cast<RankedTensorType>();
+auto input_type = op.value().getType().dyn_cast<RankedTensorType>();
5827
if (!input_type)returnfailure();
5828
if (input_type.hasStaticShape() && input_type.getNumElements() <=1)
5829
// No shuffling is required, so copy input directly to output.