experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix code style

Browse Source
This commit is contained in:
zilinzhu 2020-06-05 14:04:16 +08:00
parent b9fef3463e
commit a8dcaa2b8f
4 changed files with 11 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/core/kernels/data/experimental/map_and_batch_dataset_op.cc
View File

@ -125,8 +125,9 @@ class MapAndBatchDatasetOp::Dataset : public DatasetBase {
}
int64 Cardinality() const override {
if (!preserve_cardinality_)
if (!preserve_cardinality_) {
return kUnknownCardinality;
}
int64 n = input_->Cardinality();
if (n == kInfiniteCardinality || n == kUnknownCardinality) {
return n;

5
tensorflow/core/kernels/data/experimental/scan_dataset_op.cc
View File

@ -107,11 +107,12 @@ class ScanDatasetOp : public UnaryDatasetOpKernel {
string DebugString() const override { return "ScanDatasetOp::Dataset"; }
int64 Cardinality() const override {
if (preserve_cardinality_)
if (preserve_cardinality_) {
return input_->Cardinality();
else
} else {
return kUnknownCardinality;
}
}
Status CheckExternalState() const override {
TF_RETURN_IF_ERROR(captured_func_->CheckExternalState());

5
tensorflow/core/kernels/data/map_dataset_op.cc
View File

@ -73,11 +73,12 @@ class MapDatasetOp::Dataset : public DatasetBase {
}
int64 Cardinality() const override {
if (preserve_cardinality_)
if (preserve_cardinality_) {
return input_->Cardinality();
else
} else {
return kUnknownCardinality;
}
}
Status CheckExternalState() const override {
TF_RETURN_IF_ERROR(captured_func_->CheckExternalState());

5
tensorflow/core/kernels/data/parallel_map_dataset_op.cc
View File

@ -110,11 +110,12 @@ class ParallelMapDatasetOp::Dataset : public DatasetBase {
}
int64 Cardinality() const override {
if (preserve_cardinality_)
if (preserve_cardinality_) {
return input_->Cardinality();
else
} else {
return kUnknownCardinality;
}
}
Status CheckExternalState() const override {
TF_RETURN_IF_ERROR(captured_func_->CheckExternalState());