// initializer is given.
class InsertCallOnceOpFromSessionInitializerPass
    : public impl::InsertCallOnceOpFromSessionInitializerPassBase<
          InsertCallOnceOpFromSessionInitializerPass> {
 public:
  MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(
      InsertCallOnceOpFromSessionInitializerPass)

 private:
  void runOnOperation() override;
};

  let summary = "Calculate arithmetic count for tfl operations.";
  let constructor = "CreateGetArithmeticCountPass()";
  let dependentDialects = ["TFL::TensorFlowLiteDialect"];
}

def InsertCallOnceOpFromSessionInitializerPass : Pass<"tfl-insert-call-once-op", "mlir::ModuleOp"> {
  let summary = "Insert CallOnce op when tf_saved_model's session initializer is give.";
  let constructor = "CreateInsertCallOnceOpFromSessionInitializerPass()";