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

[MLIR][KernelGen] Add test for tf.Erf operation

Browse Source 
PiperOrigin-RevId: 352392820
Change-Id: Id0deacc1b7bf623a0d5b1f9201f8eb133787ad01
This commit is contained in:
A. Unique TensorFlower 2021-01-18 05:05:58 -08:00 committed by TensorFlower Gardener
parent e19e3d081a
commit 6a9c366ae0
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/compiler/mlir/tools/kernel_gen/tests/tf-legalize-to-lmhlo.mlir
View File

@ -26,3 +26,9 @@ func @sinh(%arg0: tensor<*xf32>) -> tensor<*xf32> {
%0 = "tf.Sinh"(%arg0) { } : (tensor<*xf32>) -> tensor<*xf32>
return %0 : tensor<*xf32>
}
func @erf(%arg0: tensor<*xf32>) -> tensor<*xf32> {
%0 = "tf.Erf"(%arg0) { } : (tensor<*xf32>) -> tensor<*xf32>
return %0 : tensor<*xf32>
}