See the License for the specific language governing permissions and
limitations under the License.
==============================================================================*/

#include "tensorflow/compiler/mlir/tfrt/function/function.h"

#include "absl/log/log.h"
#include "absl/strings/match.h"
#include "mlir/IR/OperationSupport.h"  // from @llvm-project
#include "mlir/Pass/PassManager.h"  // from @llvm-project

  // to proactively transfer the result to host since the consumer op (or
  // function) of the result may still be on TPU.
  // TODO(b/194081364): remove this option once we unify servo TPU serving
  // result transfer behavior.
  bool tpu_transfer_result_to_host = false;
};

// Compile MLIR generated by tf.function in TF dialect into BEF.
Status CompileTFMLIRToBEF(const TfrtFunctionCompileOptions& options,

# RUN: tf-mlir-translate -graphdef-to-mlir -tf-enable-shape-inference-on-import=false %s -o - | FileCheck %s

# Verify for functions with control return values, the island with a consumed
# data output and a consumed control has both its outputs added to the GraphOps
# FetchOp.

# Match the island containing the "tf.Neg", capture the output
# CHECK:          %[[ISLAND:[a-z_0-9]*]], %[[ISLAND_control:[a-z_0-9]*]] = tf_executor.island wraps "tf.Neg"

    tf_executor.fetch
  }
  func.return
}

// CHECK:      library {
// CHECK:        function {
// CHECK-NEXT:     signature {
// CHECK-NEXT:       name: "bar"
// CHECK-NEXT:     }
// CHECK:          node_def {
// CHECK-NEXT:       name: "tf.Const"
// CHECK-NEXT:       op: "Const"
// CHECK-NEXT:       attr {

// RUN: tf-opt %s -tf-functional-control-flow-to-cfg -split-input-file | FileCheck %s

func.func private @testIf1Then(tensor<*xf32>, tensor<*xf32>) -> tensor<*xf32>
func.func private @testIf1Else(tensor<*xf32>, tensor<*xf32>) -> tensor<*xf32>

// CHECK-LABEL: func @testIf1Result(%arg0: tensor<i1>, %arg1: tensor<*xf32>, %arg2: tensor<*xf32>)
func.func @testIf1Result(tensor<i1>, tensor<*xf32>, tensor<*xf32>) -> tensor<*xf32> {

          BLOCK type=kotlin.Unit origin=null
            CALL 'public final fun forEach <T> (action: kotlin.Function1<T of kotlin.collections.forEach, kotlin.Unit>): kotlin.Unit [inline] declared in kotlin.collections' type=kotlin.Unit origin=null
              <T>: kotlin.Int

Tres Popp <******@****.***> 1648211118 -0700

    FUN name:PostCardSimple visibility:public modality:FINAL <> (navigateToArticle:kotlin.Function1<kotlin.String, kotlin.Unit>, isFavorite:kotlin.Boolean, onToggleFavorite:kotlin.Function0<kotlin.Unit>) returnType:kotlin.Unit
      annotations:
        MyComposable
      VALUE_PARAMETER name:navigateToArticle index:0 type:kotlin.Function1<kotlin.String, kotlin.Unit>
      VALUE_PARAMETER name:isFavorite index:1 type:kotlin.Boolean

  }
}
node {
  name: "val"
  op: "Placeholder"
  attr {
    key: "dtype"
    value {
      type: DT_FLOAT
    }
  }
  experimental_debug_info {
  }
}
library {
  function {
    signature {
      name: "cond"
      input_arg {
        name: "cond"
        type: DT_INT32
      }
      input_arg {
        name: "cond1"
        type: DT_FLOAT
      }

Lkotlin/jvm/functions/Function1;
Lkotlin/jvm/functions/Function20;
Lkotlin/jvm/functions/Function21;
Lkotlin/jvm/functions/Function22;
Lkotlin/jvm/functions/Function2;
Lkotlin/jvm/functions/Function3;
Lkotlin/jvm/functions/Function4;
Lkotlin/jvm/functions/Function5;
Lkotlin/jvm/functions/Function6;
Lkotlin/jvm/functions/Function7;
Lkotlin/jvm/functions/Function8;
Lkotlin/jvm/functions/Function9;
Lkotlin/jvm/internal/ArrayIterator;