// license that can be found in the LICENSE file.

//go:build ignore
// +build ignore

// Savedir archives a directory tree as a txtar archive printed to standard output.
//
// Usage:
//
//	go run savedir.go /path/to/dir >saved.txt
//
// Typically the tree is later extracted during a test with tg.extract("testdata/saved.txt").
package main

import (
	"cmd/go/internal/str"
	"flag"
	"fmt"
	"internal/txtar"

// CHECK: "tf.SaveV2"(%[[ARG]], %[[CONST_0]], %[[CONST_1]], %[[READ_VARIABLE_0]], %[[READ_VARIABLE_1]])
// CHECK: return
}

// -----


// SaveV2 op not created when SessionInitializerOp doesn't exist.

module attributes {tf_saved_model.semantics} {
// CHECK-NOT: @tf_quant__save
}

// -----

// SaveV2 op not created when there are no VarHandleOp in the session

  init_op = variables.global_variables_initializer()
  saver = saver_lib.Saver(write_version=saver_pb2.SaverDef.V1)
  with session.Session() as sess:
    sess.run(init_op)
    sess.run(y.assign(y + 42))
    # Without the checkpoint, the variable won't be set to 42.
    ckpt = os.path.join(out_dir, 'test_graph_tfadd_with_ckpt.ckpt')
    saver.save(sess, ckpt)


def tfadd_with_ckpt_saver(out_dir):

    return "Inserts a new function that wraps a SaveV2 op. The SaveV2 op saves "
           "the values of the VarHandleOps that are found in the initializer "
           "function of 'restore_op' type.";
  }

  void runOnOperation() override;
};

// Finds `tf.AssignVariableOp(tf.VarHandleOp, tf.Const)` patterns and removes
// `tf.AssignVariableOp`s and `tf.Const`s. Collects and returns the

// CHECK: %[[CTL_0:.*]] = tf_executor.island wraps "tf.SaveV2"(%[[ARG_0]], %[[CST_0]], %[[CST_1]], %[[READ_VARIABLE]]) : (tensor<!tf_type.string>, tensor<1x!tf_type.string>, tensor<1x!tf_type.string>, tensor<2xf32>) -> ()

// Test that the Identity op has been created to fetch the file prefix
// argument. It should also have control dependency to the `SaveV2` op.

  return op_ptr->Execute(tensors, &num_retvals);
}

// Op: SaveV2()
// Summary: Saves tensors in V2 checkpoint format.
//
// Description:
//   By default, saves the named tensors in full.  If the caller wishes to save
//   specific slices of full tensors, "shape_and_slices" should be non-empty
//   strings and correspondingly well-formed.
Status SaveV2(AbstractContext* ctx, AbstractTensorHandle* const prefix,

#define struct union /* Great space saver */...

# RUN: tf-mlir-translate -graphdef-to-splatted-mlir %s -o - | FileCheck %s

# CHECK: tf.Const
# CHECK-SAME: value = dense<""> : tensor<3x!tf_type.string>

node {
  name: "save/SaveV2/shape_and_slices"
  op: "Const"
  attr {
    key: "_output_shapes"
    value {
      list {
        shape {
          dim {
            size: 3
          }
        }
      }
    }
  }
  attr {
    key: "dtype"
    value {

#define struct union /* Great space saver */...

#define struct union /* Great space saver */...