// has been caled
type fakeOGCounter struct {
	// calledFuncs stores name and count of functions
	calledFuncs map[string]int
	opFunc      func() volumetypes.OperationContext
}

var _ OperationGenerator = &fakeOGCounter{}

// NewFakeOGCounter returns a OperationGenerator
func NewFakeOGCounter(opFunc func() volumetypes.OperationContext) OperationGenerator {
	return &fakeOGCounter{
		calledFuncs: map[string]int{},
		opFunc:      opFunc,
	}

    func.return %0 : tensor<i32>
  }
  func.func @while_cond_without_cluster_attr(%arg0: tensor<i32>) -> tensor<i1> {
    %0 = "tf.PartitionedCallOp"(%arg0) { f = @callee_func} : (tensor<i32>) -> tensor<i1>
    func.return %0 : tensor<i1>
  }
  func.func @callee_func(%arg0: tensor<i32>) -> tensor<i1> {
    %0 = "tf.some_op"(%arg0) : (tensor<i32>) -> tensor<i1>
    func.return %0 : tensor<i1>
  }

      SymbolTable::getSymbolUses(target_func, &module_op.getBodyRegion());
  for (auto& function_use : function_uses.value()) {
    auto call_op = function_use.getUser();
    auto caller_func = call_op->getParentOfType<func::FuncOp>();
    if (!caller_func) return failure();

    builder.setInsertionPoint(call_op);
    for (auto [lifted_op, arg_idx] : lifted_op_and_arg_idx) {
      auto new_op = builder.clone(*lifted_op, mapping);

    }
    func.func @while_cond_without_cluster_attr(%arg0: tensor<i32>) -> tensor<i1> {
      %0 = "tf.some_other_op"(%arg0) : (tensor<i32>) -> tensor<i1>
      func.return %0 : tensor<i1>
    }
    func.func @callee_func(%arg0: tensor<i32>) -> tensor<i1> {
      %0 = "tf.some_op"(%arg0) : (tensor<i32>) -> tensor<i1>
      func.return %0 : tensor<i1>
    }
  }

func.func @test_arg_count_mismatch(%arg0: tensor<i32>) -> () {
  // expected-error @below {{argument count mismatch: 'args' has 1 argument(s), but 'call_func' expects 2}}
  "tf.LegacyCall"(%arg0) {f = @call_func} : (tensor<i32>) -> tensor<i32>
  func.return
}

func.func @call_func(%arg0: tensor<f32>, %arg1: tensor<f32>) -> tensor<f32> {
  func.return %arg0 : tensor<f32>
}

// -----