diff --git a/src/TensorFlowNET.Core/Data/IDatasetV2.cs b/src/TensorFlowNET.Core/Data/IDatasetV2.cs index 77bc9169..0f38d531 100644 --- a/src/TensorFlowNET.Core/Data/IDatasetV2.cs +++ b/src/TensorFlowNET.Core/Data/IDatasetV2.cs @@ -57,7 +57,7 @@ namespace Tensorflow IDatasetV2 map(Func map_func, bool use_inter_op_parallelism = true, - bool preserve_cardinality = false, + bool preserve_cardinality = true, bool use_legacy_function = false); IDatasetV2 map(Func map_func, diff --git a/src/TensorFlowNET.Core/Data/MapDataset.cs b/src/TensorFlowNET.Core/Data/MapDataset.cs index c593322b..09513f94 100644 --- a/src/TensorFlowNET.Core/Data/MapDataset.cs +++ b/src/TensorFlowNET.Core/Data/MapDataset.cs @@ -25,7 +25,9 @@ namespace Tensorflow variant_tensor = ops.map_dataset(input_dataset.variant_tensor, func, output_types, - output_shapes); + output_shapes, + use_inter_op_parallelism: use_inter_op_parallelism, + preserve_cardinality: preserve_cardinality); } } } diff --git a/test/TensorFlowNET.UnitTest/Dataset/DatasetTest.cs b/test/TensorFlowNET.UnitTest/Dataset/DatasetTest.cs index 417dac09..746fea84 100644 --- a/test/TensorFlowNET.UnitTest/Dataset/DatasetTest.cs +++ b/test/TensorFlowNET.UnitTest/Dataset/DatasetTest.cs @@ -143,7 +143,7 @@ namespace TensorFlowNET.UnitTest.Dataset } } - [TestMethod, Ignore] + [TestMethod] public void Cardinality() { var dataset = tf.data.Dataset.range(10);