Home
last modified time | relevance | path

Searched refs:preserve_cardinality_ (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dmap_dataset_op.cc50 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()
Dparallel_map_dataset_op.cc86 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()
Dparallel_map_dataset_op_test.cc44 preserve_cardinality_(preserve_cardinality) { in ParallelMapDatasetParams()
80 {ParallelMapDatasetOp::kPreserveCardinality, preserve_cardinality_}}; in GetAttributes()
98 bool preserve_cardinality_; member in tensorflow::data::__anon7368ffe00111::ParallelMapDatasetParams
Dmap_dataset_op.h49 bool preserve_cardinality_; variable
Dparallel_map_dataset_op.h55 bool preserve_cardinality_; variable
Ddataset_test_base.h258 preserve_cardinality_(preserve_cardinality) { in MapDatasetParams()
281 bool preserve_cardinality_; variable
Ddataset_test_base.cc1055 {MapDatasetOp::kPreserveCardinality, preserve_cardinality_}}; in GetAttributes()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dscan_dataset_op.cc48 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
Dmap_and_batch_dataset_op.cc98 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()
Dmap_and_batch_dataset_op_test.cc42 preserve_cardinality_(preserve_cardinality) { in MapAndBatchDatasetParams()
79 {MapAndBatchDatasetOp::kPreserveCardinality, preserve_cardinality_}}; in GetAttributes()
97 bool preserve_cardinality_; member in tensorflow::data::experimental::__anon402561240111::MapAndBatchDatasetParams
Dmap_and_batch_dataset_op.h54 bool preserve_cardinality_; variable