Browse Source

Fix dataset_cardinality for dataset map.

tags/keras_v0.3.0
Oceania2018 4 years ago
parent
commit
878226b416
3 changed files with 5 additions and 3 deletions
  1. +1
    -1
      src/TensorFlowNET.Core/Data/IDatasetV2.cs
  2. +3
    -1
      src/TensorFlowNET.Core/Data/MapDataset.cs
  3. +1
    -1
      test/TensorFlowNET.UnitTest/Dataset/DatasetTest.cs

+ 1
- 1
src/TensorFlowNET.Core/Data/IDatasetV2.cs View File

@@ -57,7 +57,7 @@ namespace Tensorflow

IDatasetV2 map(Func<Tensor, Tensor> map_func,
bool use_inter_op_parallelism = true,
bool preserve_cardinality = false,
bool preserve_cardinality = true,
bool use_legacy_function = false);

IDatasetV2 map(Func<Tensor, (Tensor, Tensor), (Tensor, Tensor)> map_func,


+ 3
- 1
src/TensorFlowNET.Core/Data/MapDataset.cs View File

@@ -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);
}
}
}

+ 1
- 1
test/TensorFlowNET.UnitTest/Dataset/DatasetTest.cs View File

@@ -143,7 +143,7 @@ namespace TensorFlowNET.UnitTest.Dataset
}
}

[TestMethod, Ignore]
[TestMethod]
public void Cardinality()
{
var dataset = tf.data.Dataset.range(10);


Loading…
Cancel
Save