diff --git a/src/TensorFlowNET.Core/Operations/TensorArray.cs b/src/TensorFlowNET.Core/Tensors/TensorArray.cs similarity index 94% rename from src/TensorFlowNET.Core/Operations/TensorArray.cs rename to src/TensorFlowNET.Core/Tensors/TensorArray.cs index 60e1bde5..f84072a8 100644 --- a/src/TensorFlowNET.Core/Operations/TensorArray.cs +++ b/src/TensorFlowNET.Core/Tensors/TensorArray.cs @@ -17,8 +17,9 @@ using System; using System.Collections.Generic; using System.Text; +using Tensorflow.Operations; -namespace Tensorflow.Operations +namespace Tensorflow { /// /// TensorArray is designed to hide an underlying implementation object @@ -29,9 +30,9 @@ namespace Tensorflow.Operations /// `while_loop` and `map_fn`. It supports gradient back-propagation via special /// "flow" control flow dependencies. /// - public class TensorArray + public class TensorArray : ITensorOrTensorArray { - _GraphTensorArray _implementation; + internal _GraphTensorArray _implementation; public TF_DataType dtype => _implementation._dtype; public Tensor handle => _implementation._handle;