using System;

using System.Collections.Generic;

using System.Linq;

using static Tensorflow.Binding;

namespace Tensorflow.Graphs

{

/// <summary>

/// Graph representing a function body.

/// </summary>

public class FuncGraph : Graph

{

List<Operation> inputs;

List<Operation> outputs;

Graph outer_graph;

string func_name;

IntPtr func_handle;

public string FuncName => func_name;

/// <summary>

/// Construct a new FuncGraph.

/// </summary>

public FuncGraph(string name) : base()

{

outer_graph = ops.get_default_graph();

func_name = name;

tf.Context.graph_mode();

as_default();

}

public IntPtr ToGraph(Operation[] opers,

Operation[] inputs, Operation[] outputs,

string[] output_names)

{

using var status = new Status();

func_handle = c_api.TF_GraphToFunction(_handle,

func_name,

false,

opers.Length,

opers.Select(x => (IntPtr)x).ToArray(),

inputs.Length,

inputs.Select(x => new TF_Output(x, 0)).ToArray(),

outputs.Length,

outputs.Select(x => new TF_Output(x, 0)).ToArray(),

output_names == null || output_names.Length == 0 ? null : output_names,

IntPtr.Zero,

null,

status.Handle);

status.Check(true);

c_api.TF_GraphCopyFunction(outer_graph, func_handle, IntPtr.Zero, status.Handle);

status.Check(true);

c_api.TFE_ContextAddFunction(tf.Context.Handle, func_handle, status.Handle);

status.Check(true);

func_name = c_api.StringPiece(c_api.TF_FunctionName(func_handle));

return func_handle;

}

protected override void DisposeManagedResources()

{

base.DisposeManagedResources();

tf.Context.restore_mode();

}

}

}