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

Fix op definition templates for isinf and isnan.

Browse Source 
They return a i1 tensor.

PiperOrigin-RevId: 339840456
Change-Id: I7b25d8b7c370f31a7c794117912e0e27636eee36
This commit is contained in:
Adrian Kuegel 2020-10-30 03:11:04 -07:00 committed by TensorFlower Gardener
parent 5234e72159
commit be084bd7a4
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/core/kernels/mlir_generated/op_definitions/isinf.mlir.tmpl
View File

@ -1,5 +1,5 @@
func @Isinf_elem_type(%arg0: tensor<*xelem_type>)
-> tensor<*xelem_type> attributes {tf_entry, llvm.emit_c_interface} {
-> tensor<*xi1> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.IsInf"(%arg0) : (tensor<*xelem_type>) -> tensor<*xi1>
return %0 : tensor<*xi1>
}

2
tensorflow/core/kernels/mlir_generated/op_definitions/isnan.mlir.tmpl
View File

@ -1,5 +1,5 @@
func @Isnan_elem_type(%arg0: tensor<*xelem_type>)
-> tensor<*xelem_type> attributes {tf_entry, llvm.emit_c_interface} {
-> tensor<*xi1> attributes {tf_entry, llvm.emit_c_interface} {
%0 = "tf.IsNan"(%arg0) : (tensor<*xelem_type>) -> tensor<*xi1>
return %0 : tensor<*xi1>
}