experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[MLIR][KernelGen] Add E2E test for tf.Sin

Browse Source 
Organize E2E test such that it can be be extended by other `tf` operations.

PiperOrigin-RevId: 333039645
Change-Id: I20fa2befda808d4943df0a3beb402831362fc688
This commit is contained in:
A. Unique TensorFlower 2020-09-22 03:01:27 -07:00 committed by TensorFlower Gardener
parent d8ec6ffe96
commit a0a4b95f50
1 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/compiler/mlir/tools/kernel_gen/transforms/bufferize.cc
View File

@ -159,14 +159,10 @@ class TensorCastOpConverter
LogicalResult matchAndRewrite(
TensorCastOp op, ArrayRef<Value> operands,
ConversionPatternRewriter &rewriter) const final {
auto tensor_ty = op.getType().dyn_cast<RankedTensorType>();
if (!tensor_ty) return failure();
Value arg = operands.front();
auto arg_ty = arg.getType().dyn_cast<MemRefType>();
if (!arg_ty) return failure();
if (!arg.getType().isa<MemRefType>()) return failure();
auto result_ty = converter->convertType(tensor_ty);
auto result_ty = converter->convertType(op.getType());
rewriter.replaceOpWithNewOp<MemRefCastOp>(op, arg, result_ty);
return success();