using System;

using System.Collections.Generic;

using System.Linq;

namespace Tensorflow.Keras.Engine

{

public partial class Layer

{

public virtual List<ILayer> Layers => _self_tracked_trackables;

protected void StackLayers(params ILayer[] layers)

{

_self_tracked_trackables.AddRange(layers);

}

public virtual Shape ComputeOutputShape(Shape input_shape)

=> throw new NotImplementedException("");

protected List<IVariableV1> _gather_children_variables(bool include_trainable = false, bool include_non_trainable = false)

{

List<IVariableV1> res = new();

var nested_layers = _flatten_layers(false, false);

foreach (var layer in nested_layers)

{

if (layer is Layer l)

{

if (include_trainable == true && include_non_trainable == true)

{

res.AddRange(l.Variables);

}

else if (include_trainable == true && include_non_trainable == false)

{

res.AddRange(l.TrainableVariables);

}

else if(include_trainable == false && include_non_trainable == true)

{

res.AddRange(l.NonTrainableVariables);

}

}

}

return res;

}

}

}