/external/tensorflow/tensorflow/core/kernels/data/ |
D | map_dataset_op.cc | 50 preserve_cardinality_(preserve_cardinality), in Dataset() 76 if (preserve_cardinality_) { in Cardinality() 120 b->BuildAttrValue(preserve_cardinality_, &preserve_cardinality_attr); in AsGraphDefInternal() 164 if (dataset()->preserve_cardinality_) { in GetNextInternal() 208 const bool preserve_cardinality_; member in tensorflow::data::MapDatasetOp::Dataset 224 ctx->GetAttr(kPreserveCardinality, &preserve_cardinality_)); in MapDatasetOp() 235 output_shapes_, preserve_cardinality_); in MakeDataset()
|
D | parallel_map_dataset_op.cc | 86 preserve_cardinality_(preserve_cardinality), in Dataset() 116 if (preserve_cardinality_) { in Cardinality() 188 b->BuildAttrValue(preserve_cardinality_, &preserve_cardinality_attr); in AsGraphDefInternal() 211 preserve_cardinality_(params.dataset->preserve_cardinality_), in Iterator() 485 if (preserve_cardinality_) { in ProcessResult() 642 const bool preserve_cardinality_; member in tensorflow::data::ParallelMapDatasetOp::Dataset::Iterator 670 const bool preserve_cardinality_; member in tensorflow::data::ParallelMapDatasetOp::Dataset 701 ctx->GetAttr(kPreserveCardinality, &preserve_cardinality_)); in ParallelMapDatasetOp() 733 preserve_cardinality_, op_version_); in MakeDataset()
|
D | parallel_map_dataset_op_test.cc | 44 preserve_cardinality_(preserve_cardinality) { in ParallelMapDatasetParams() 80 {ParallelMapDatasetOp::kPreserveCardinality, preserve_cardinality_}}; in GetAttributes() 98 bool preserve_cardinality_; member in tensorflow::data::__anon7368ffe00111::ParallelMapDatasetParams
|
D | map_dataset_op.h | 49 bool preserve_cardinality_; variable
|
D | parallel_map_dataset_op.h | 55 bool preserve_cardinality_; variable
|
D | dataset_test_base.h | 258 preserve_cardinality_(preserve_cardinality) { in MapDatasetParams() 281 bool preserve_cardinality_; variable
|
D | dataset_test_base.cc | 1055 {MapDatasetOp::kPreserveCardinality, preserve_cardinality_}}; in GetAttributes()
|
/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
D | scan_dataset_op.cc | 48 ctx, ctx->GetAttr("preserve_cardinality", &preserve_cardinality_)); in ScanDatasetOp() 67 output_shapes_, preserve_cardinality_, use_default_device_); in MakeDataset() 87 preserve_cardinality_(preserve_cardinality), in Dataset() 110 if (preserve_cardinality_) { in Cardinality() 152 b->BuildAttrValue(preserve_cardinality_, &preserve_cardinality_attr); in AsGraphDefInternal() 239 if (dataset()->preserve_cardinality_) { in GetNextInternal() 310 const bool preserve_cardinality_; member in tensorflow::data::experimental::__anon700576d90111::ScanDatasetOp::Dataset 318 bool preserve_cardinality_; member in tensorflow::data::experimental::__anon700576d90111::ScanDatasetOp
|
D | map_and_batch_dataset_op.cc | 98 preserve_cardinality_(preserve_cardinality), in Dataset() 127 if (!preserve_cardinality_) { in Cardinality() 169 b->BuildAttrValue(preserve_cardinality_, &preserve_cardinality_attr); in AsGraphDefInternal() 420 if (dataset()->preserve_cardinality_ && errors::IsOutOfRange(status)) { in CallFunction() 687 const bool preserve_cardinality_; member in tensorflow::data::experimental::MapAndBatchDatasetOp::Dataset 698 ctx->GetAttr(kPreserveCardinality, &preserve_cardinality_)); in MapAndBatchDatasetOp() 730 std::move(captured_func), preserve_cardinality_); in MakeDataset()
|
D | map_and_batch_dataset_op_test.cc | 42 preserve_cardinality_(preserve_cardinality) { in MapAndBatchDatasetParams() 79 {MapAndBatchDatasetOp::kPreserveCardinality, preserve_cardinality_}}; in GetAttributes() 97 bool preserve_cardinality_; member in tensorflow::data::experimental::__anon402561240111::MapAndBatchDatasetParams
|
D | map_and_batch_dataset_op.h | 54 bool preserve_cardinality_; variable
|