# This script generates a jpg image of our internal dependency graph.
# It relies on go mod and dot to do so and should be run from K8s root.
# TODO: Containerize the script to remove dependency issues with go mod and dot.

# To generate graph with all Kubernetes modules run
# ./hack/module-graph.sh
# To generate graph with just staging modules run
# ./hack/module-graph.sh staging

error_exit()
{
	echo "$1" 1>&2
	exit 1
}

// RUN: tf-mlir-translate -mlir-tf-graph-to-hlo-text %s -tf-input-shapes=2:2 -tf-input-data-types=DT_FLOAT,DT_FLOAT -tf-xla-input-types=parameter,resource -tf-xla-emit-return-tuple | FileCheck %s

module attributes {tf.versions = {producer = 511 : i32}} {
  func.func @main(%arg0: tensor<*xf32>, %arg1: tensor<*x!tf_type.resource>) {
    tf_executor.graph {

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

this is not a valid graph def

# CHECK: Error parsing Protobuf

# RUN: tf-mlir-translate -graphdef-to-mlir -tf-enable-shape-inference-on-import=false %s -tf-graph-as-function -tf-control-output-arrays=assign_variable | tf-mlir-translate -mlir-tf-graph-to-hlo-text -tf-input-shapes=2:2 -tf-input-data-types=DT_FLOAT,DT_FLOAT -tf-xla-input-types=parameter,resource -tf-xla-emit-return-tuple | FileCheck %s

node {
  name: "arg0"
  op: "_Arg"
  attr {
    key: "T"
    value {
      type: DT_FLOAT
    }
  }
  attr {

    tf_executor.fetch %1#0, %2#0 : tensor<f32>, tensor<f32>
  }
  func.return %graph#0, %graph#1 : tensor<f32>, tensor<f32>
}

func.func @function0(%arg0: tensor<*xf32>, %arg1: tensor<*x!tf_type.resource>) -> tensor<*xf32>
attributes {tf.signature.is_stateful} {
  %graph = tf_executor.graph {

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

// dot -Tpng src/docs/userguide/img/xctest-task-graph.dot > src/docs/userguide/img/xctest-task-graph.png
digraph xctestTaskGraph {
  graph [ dpi = 100, fontname="Sans"];
  node [fontname = "Sans"];
  edge [fontname = "Sans"];
  node [shape=rectangle, fixedsize=true, width=2.5, height=0.5];

	virtual     bool
	virtualLock sync.RWMutex
	// when processing an Update event, we need to compare the updated
	// ownerReferences with the owners recorded in the graph.
	owners []metav1.OwnerReference
}

// clone() must only be called from the single-threaded GraphBuilder.processGraphChanges()
func (n *node) clone() *node {
	c := &node{
		identity:           n.identity,

# RUN: tf-mlir-translate -graphdef-to-mlir -tf-enable-shape-inference-on-import=false %s -tf-graph-as-function -tf-xla-compile-device-type="GPU" -tf-enable-soft-placement-on-import=true -o - | FileCheck %s

# Verify main graph was converted to a function, args/rets are mapped correctly,
# and ops in the main graph are retained. In addition, check if subsequent
# functions are converted.

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

// dot -Tpng src/docs/userguide/img/xcode-task-graph.dot > src/docs/userguide/img/xcode-task-graph.png
digraph xcodeTaskGraph {
  graph [ dpi = 100, fontname="Sans"];
  node [fontname = "Sans"];
  edge [fontname = "Sans"];
  node [shape=rectangle, fixedsize=true, width=3.3, height=0.5];

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

// dot -Tpng src/docs/userguide/img/swift-application-task-graph.dot > src/docs/userguide/img/swift-application-task-graph.png
digraph swiftApplicationTaskGraph {
  graph [ dpi = 100, fontname="Sans"];
  node [fontname = "Sans"];
  edge [fontname = "Sans"];
  node [shape=rectangle, fixedsize=true, width=2.5, height=0.5];