colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[MLIR][KernelGen] Add E2E tests for `tf.Tan` and `tf.Sinh` 

PiperOrigin-RevId: 333275089
This commit is contained in:
A. Unique TensorFlower 2020-09-23 05:50:02 -07:00 committed by TensorFlow MLIR Team
parent be409a2409
commit 08514eaa5f
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops.td
View File

@ -477,7 +477,8 @@ def ReshapeMemRefCastOp: Op<LHLO_Dialect, "reshape_memref_cast", [
let results = (outs AnyRankedOrUnrankedMemRef:$result);
let extraClassDeclaration = [{
MemRefType getType() { return getResult().getType().cast<MemRefType>(); }
BaseMemRefType getType() {
return getResult().getType().cast<BaseMemRefType>(); }
}];
let verifier = [{ return Verify(*this); }];