Home
last modified time | relevance | path

Searched refs:op_version_ (Results 1 – 25 of 32) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/data/
Dbatch_dataset_op.cc63 op_version_(op_version), in Dataset()
92 params.op_version = op_version_; in MakeIteratorInternal()
107 params.op_version = op_version_; in DebugString()
251 const int op_version_; member in tensorflow::data::BatchDatasetOp::Dataset
258 op_version_(ctx->def().op() == kBatchDataset ? 1 : 2) { in BatchDatasetOp()
272 if (op_version_ > 1) { in MakeDataset()
278 op_version_); in MakeDataset()
Doptimize_dataset_op.cc57 op_version_ = 1; in OptimizeDatasetOp()
59 op_version_ = 2; in OptimizeDatasetOp()
68 if (op_version_ == 1) { in MakeDataset()
71 } else if (op_version_ == 2) { in MakeDataset()
Dparallel_map_dataset_op.cc88 op_version_(op_version) { in Dataset()
97 params.op_version = op_version_; in MakeIteratorInternal()
110 params.op_version = op_version_; in DebugString()
149 if (op_version_ == 1) { in AsGraphDefInternal()
174 if (op_version_ == 1) { in AsGraphDefInternal()
180 if (op_version_ == 2) { in AsGraphDefInternal()
672 const int op_version_; member in tensorflow::data::ParallelMapDatasetOp::Dataset
676 : UnaryDatasetOpKernel(ctx), op_version_(ctx->HasAttr(kSloppy) ? 1 : 2) { in ParallelMapDatasetOp()
684 if (op_version_ == 1) { in ParallelMapDatasetOp()
694 if (op_version_ == 2) { in ParallelMapDatasetOp()
[all …]
Dshuffle_dataset_op.h54 int op_version_ = 0; variable
72 int op_version_ = 0; variable
Dpadded_batch_dataset_op.cc62 op_version_(op_version), in Dataset()
94 params.op_version = op_version_; in MakeIteratorInternal()
109 params.op_version = op_version_; in DebugString()
389 const int op_version_; member in tensorflow::data::PaddedBatchDatasetOp::Dataset
396 op_version_(ctx->def().op() == "PaddedBatchDataset" ? 1 : 2) { in PaddedBatchDatasetOp()
410 if (op_version_ > 1) { in MakeDataset()
458 input, op_version_); in MakeDataset()
Dfixed_length_record_dataset_op.cc63 op_version_(op_version) {} in Dataset()
68 params.op_version = op_version_; in MakeIteratorInternal()
92 params.op_version = op_version_; in DebugString()
454 const int op_version_; member in tensorflow::data::FixedLengthRecordDatasetOp::Dataset
460 op_version_(ctx->def().op() == kFixedLengthRecordDataset ? 1 : 2) {} in FixedLengthRecordDatasetOp()
504 if (op_version_ > 1) { in MakeDataset()
514 footer_bytes, buffer_size, compression_type, op_version_); in MakeDataset()
Dshuffle_dataset_op.cc619 op_version_ = 3; in ShuffleDatasetOp()
621 op_version_ = 2; in ShuffleDatasetOp()
623 op_version_ = 1; in ShuffleDatasetOp()
645 if (op_version_ == 3) { in MakeDataset()
682 } else if (op_version_ == 2) { in MakeDataset()
712 if (op_version_ != 1) { in MakeDataset()
714 << op_version_ << ". Defaulting to version 1."; in MakeDataset()
870 op_version_ = 2; in ShuffleAndRepeatDatasetOp()
872 op_version_ = 1; in ShuffleAndRepeatDatasetOp()
909 if (op_version_ == 2) { in MakeDataset()
[all …]
Ddataset_ops.cc52 : OpKernel(ctx), op_version_(ctx->def().op() == kDatasetToGraph ? 1 : 2) { in DatasetToGraphOp()
53 if (op_version_ == 2) { in DatasetToGraphOp()
Dparallel_interleave_dataset_op.cc180 op_version_(op_version), in Dataset()
198 params.op_version = op_version_; in MakeIteratorInternal()
217 params.op_version = op_version_; in DebugString()
266 if (op_version_ >= 4) { in AsGraphDefInternal()
292 if (op_version_ == 2) { in AsGraphDefInternal()
297 if (op_version_ >= 3) { in AsGraphDefInternal()
1566 const int op_version_; member in tensorflow::data::ParallelInterleaveDatasetOp::Dataset
1573 op_version_(OpVersionFromOpName(ctx->def().op())) { in ParallelInterleaveDatasetOp()
1578 if (op_version_ == 2) { in ParallelInterleaveDatasetOp()
1588 if (op_version_ >= 3) { in ParallelInterleaveDatasetOp()
[all …]
Dfixed_length_record_dataset_op.h40 const int op_version_; variable
Dbatch_dataset_op.h41 const int op_version_; variable
Dcache_dataset_ops.h46 const int op_version_; variable
Dpadded_batch_dataset_op.h44 const int op_version_; variable
Dparallel_map_dataset_op.h50 const int op_version_; variable
Dparallel_interleave_dataset_op.h52 const int op_version_; variable
Ddataset_ops.h43 const int op_version_;
Doptimize_dataset_op.h57 int op_version_ = 0; variable
Dparallel_map_dataset_op_test.cc46 op_version_ = kOpVersion; in ParallelMapDatasetParams()
48 params.op_version = op_version_; in ParallelMapDatasetParams()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dparse_example_dataset_op.cc52 op_version_(ctx->HasAttr("deterministic") ? 2 : 1) { in ParseExampleDatasetOp()
61 if (op_version_ == 1) { in ParseExampleDatasetOp()
71 if (op_version_ == 2) { in ParseExampleDatasetOp()
206 ragged_value_types_, ragged_split_types_, op_version_); in MakeDataset()
244 op_version_(op_version) { in Dataset()
253 params.op_version = op_version_; in MakeIteratorInternal()
268 params.op_version = op_version_; in DebugString()
326 if (op_version_ == 1) { in AsGraphDefInternal()
331 if (op_version_ == 2) { in AsGraphDefInternal()
925 const int op_version_; member in tensorflow::data::experimental::__anond1cd35330111::ParseExampleDatasetOp::Dataset
[all …]
Dparallel_interleave_dataset_op.cc120 op_version_(op_version) {
129 params.op_version = op_version_; in MakeIteratorInternal()
146 params.op_version = op_version_; in DebugString()
186 if (op_version_ == 1) { in AsGraphDefInternal()
209 if (op_version_ == 2) { in AsGraphDefInternal()
1169 const int op_version_; member in tensorflow::data::experimental::ParallelInterleaveDatasetOp::Dataset
1175 op_version_(ctx->HasAttr(kDeterministic) ? 2 : 1) { in ParallelInterleaveDatasetOp()
1178 if (op_version_ == 2) { in ParallelInterleaveDatasetOp()
1201 if (op_version_ == 1) { in MakeDataset()
1234 op_version_); in MakeDataset()
Dcsv_dataset_op.cc33 op_version_(ctx->def().op() == "CSVDatasetV2" ? 2 : 1) { in CSVDatasetOp()
68 if (op_version_ > 1) { in MakeDataset()
160 std::move(na_value), op_version_); in MakeDataset()
184 op_version_(op_version), in Dataset()
240 if (op_version_ > 1) { in AsGraphDefInternal()
913 const int op_version_; member in tensorflow::data::experimental::__anon30b914390111::CSVDatasetOp::Dataset
919 const int op_version_; member in tensorflow::data::experimental::__anon30b914390111::CSVDatasetOp
Ddata_service_dataset_op.cc97 op_version_(op_version), in Dataset()
181 if (op_version_ == 2) { in AsGraphDefInternal()
837 const int op_version_; member in tensorflow::data::DataServiceDatasetOp::Dataset
872 op_version_ = 1; in DataServiceDatasetOp()
874 op_version_ = 2; in DataServiceDatasetOp()
909 if (op_version_ >= 2) { in MakeDataset()
964 *output = new Dataset(ctx, op_version_, dataset_id, processing_mode, address, in MakeDataset()
Dparallel_interleave_dataset_op.h55 const int op_version_; variable
Ddata_service_dataset_op.h77 int op_version_; variable
/external/tensorflow/tensorflow/core/kernels/
Dexample_parsing_ops.cc51 : OpKernel(ctx), op_version_(ctx->def().op() == kParseExampleV2 ? 2 : 1) { in ParseExampleOp()
52 OP_REQUIRES_OK(ctx, attrs_.Init(ctx, op_version_)); in ParseExampleOp()
66 if (op_version_ == 2) { in Compute()
131 if (op_version_ == 2) { in CheckInputShapes()
150 if (op_version_ == 2) { in CheckInputShapes()
273 if (op_version_ == 2) { in WriteOutput()
287 int op_version_; member in tensorflow::ParseExampleOp
398 op_version_(ctx->def().op() == kParseSequenceExampleV2 ? 2 : 1) { in ParseSequenceExampleOp()
399 OP_REQUIRES_OK(ctx, attrs_.Init(ctx, op_version_)); in ParseSequenceExampleOp()
422 if (op_version_ == 2) { in Compute()
[all …]

12