xref: /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/ir/testdata/test.mlir (revision b6fb3261f9314811a0f4371741dbb8839866f948)
1func.func @test(%ch: !tfrt.chain, %arg0: !corert.tensorhandle, %arg1_th: !corert.tensorhandle) {
2  %cpu = corert.get_op_handler %ch "cpu"
3  %0 = corert.executeop(%cpu) "tf.Relu"(%arg0) { T = f32 } : 1
4  %arg1 = tfrt_fallback_async.corert_tensorhandle_to_fallback_tensor %arg1_th {_tfrt_cost = 1 : i64, device = "/CPU:0"} : (!corert.tensorhandle) -> (!tfrt_fallback.tf_tensor)
5  %1 = tfrt_fallback_async.executeop key(0) cost(100) device("/CPU:0") "tf.Relu"(%arg1) { T = f32 } : 1
6  tfrt.return
7}
8