Home
last modified time | relevance | path

Searched refs:ParseScalarArgument (Results 1 – 25 of 43) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/data/
Drandom_seed_ops.cc71 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, kSeed, &seed)); in Compute()
73 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, kSeed2, &seed2)); in Compute()
77 OP_REQUIRES_OK(ctx, ParseScalarArgument<bool>(ctx, kReshuffle, &reshuffle_)); in Compute()
Dfixed_length_record_dataset_op.cc479 ParseScalarArgument<int64>(ctx, kHeaderBytes, &header_bytes)); in MakeDataset()
485 ParseScalarArgument<int64>(ctx, kRecordBytes, &record_bytes)); in MakeDataset()
491 ParseScalarArgument<int64>(ctx, kFooterBytes, &footer_bytes)); in MakeDataset()
497 ParseScalarArgument<int64>(ctx, kBufferSize, &buffer_size)); in MakeDataset()
505 OP_REQUIRES_OK(ctx, ParseScalarArgument<tstring>(ctx, kCompressionType, in MakeDataset()
Dshuffle_dataset_op.cc635 ParseScalarArgument<int64>(ctx, kBufferSize, &buffer_size)); in MakeDataset()
651 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, kSeed, &seed)); in MakeDataset()
653 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, kSeed2, &seed2)); in MakeDataset()
717 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, kSeed, &seed)); in MakeDataset()
719 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, kSeed2, &seed2)); in MakeDataset()
885 ParseScalarArgument<int64>(ctx, kBufferSize, &buffer_size)); in MakeDataset()
891 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, kSeed, &seed)); in MakeDataset()
894 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, kSeed2, &seed2)); in MakeDataset()
897 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, kCount, &count)); in MakeDataset()
Drange_dataset_op.cc306 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, kStart, &start)); in MakeDataset()
309 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, kStop, &stop)); in MakeDataset()
312 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, kStep, &step)); in MakeDataset()
Dwindow_dataset_op.cc397 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, kSize, &window_size)); in MakeDataset()
403 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, kShift, &window_shift)); in MakeDataset()
409 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, kStride, &window_stride)); in MakeDataset()
416 ctx, ParseScalarArgument<bool>(ctx, kDropRemainder, &drop_remainder)); in MakeDataset()
Dshard_dataset_op.cc249 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, kNumShards, &num_shards)); in MakeDataset()
256 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, kIndex, &index)); in MakeDataset()
Dbatch_dataset_op.cc267 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, kBatchSize, &batch_size)); in MakeDataset()
274 ctx, ParseScalarArgument<bool>(ctx, kDropRemainder, &drop_remainder)); in MakeDataset()
Dtext_line_dataset_op.cc248 OP_REQUIRES_OK(ctx, ParseScalarArgument<tstring>(ctx, kCompressionType, in MakeDataset()
253 ParseScalarArgument<int64>(ctx, kBufferSize, &buffer_size)); in MakeDataset()
Dtf_record_dataset_op.cc299 OP_REQUIRES_OK(ctx, ParseScalarArgument<tstring>(ctx, kCompressionType, in MakeDataset()
304 ParseScalarArgument<int64>(ctx, kBufferSize, &buffer_size)); in MakeDataset()
Dparallel_batch_dataset_op.cc531 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, kBatchSize, &batch_size)); in MakeDataset()
536 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, kNumParallelCalls, in MakeDataset()
541 ctx, ParseScalarArgument<bool>(ctx, kDropRemainder, &drop_remainder)); in MakeDataset()
Dinterleave_dataset_op.cc323 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, kCycleLength, &cycle_length)); in MakeDataset()
332 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, kBlockLength, &block_length)); in MakeDataset()
Ddataset_ops.cc123 ParseScalarArgument(ctx, kGraphDef, &graph_def_string)); in Compute()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dto_tf_record_op.cc38 Status ParseScalarArgument(OpKernelContext* ctx, in ParseScalarArgument() function in tensorflow::data::experimental::__anone30cd1e90111::ToTFRecordOp
64 ParseScalarArgument<tstring>(ctx, "filename", &filename)); in DoCompute()
66 TF_RETURN_IF_ERROR(ParseScalarArgument<tstring>(ctx, "compression_type", in DoCompute()
Ddata_service_ops.cc45 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, kAddress, &address)); in Compute()
50 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, kProtocol, &protocol)); in Compute()
Dauto_shard_dataset_op.cc48 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, kNumWorkers, &num_workers)); in MakeDataset()
53 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, kIndex, &index)); in MakeDataset()
Dsql_dataset_op.cc55 ctx, ParseScalarArgument<tstring>(ctx, "driver_name", &driver_name)); in MakeDataset()
58 OP_REQUIRES_OK(ctx, ParseScalarArgument<tstring>(ctx, "data_source_name", in MakeDataset()
62 OP_REQUIRES_OK(ctx, ParseScalarArgument<tstring>(ctx, "query", &query)); in MakeDataset()
Dsampling_dataset_op.cc231 OP_REQUIRES_OK(ctx, ParseScalarArgument<float>(ctx, kRate, &rate)); in MakeDataset()
232 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, kSeed, &seed)); in MakeDataset()
233 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, kSeed2, &seed2)); in MakeDataset()
Ddata_service_dataset_op.cc885 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, kDatasetId, &dataset_id)); in MakeDataset()
889 ctx, ParseScalarArgument(ctx, kProcessingMode, &processing_mode_str)); in MakeDataset()
895 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, kAddress, &address)); in MakeDataset()
900 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, kProtocol, &protocol)); in MakeDataset()
905 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, kJobName, &job_name)); in MakeDataset()
912 ctx, ParseScalarArgument(ctx, kConsumerIndex, &consumer_index_int)); in MakeDataset()
919 ParseScalarArgument(ctx, kNumConsumers, &num_consumers_int)); in MakeDataset()
926 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, kMaxOutstandingRequests, in MakeDataset()
Drandom_dataset_op.cc154 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, "seed", &seed)); in MakeDataset()
157 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, "seed2", &seed2)); in MakeDataset()
Dsliding_window_dataset_op.cc38 ctx, ParseScalarArgument<int64>(ctx, "window_size", &window_size)); in MakeDataset()
44 ctx, ParseScalarArgument<int64>(ctx, "window_shift", &window_shift)); in MakeDataset()
50 ctx, ParseScalarArgument<int64>(ctx, "window_stride", &window_stride)); in MakeDataset()
Dcsv_dataset_op.cc46 OP_REQUIRES_OK(ctx, ParseScalarArgument<tstring>(ctx, "compression_type", in MakeDataset()
81 ctx, ParseScalarArgument<int64>(ctx, "buffer_size", &buffer_size)); in MakeDataset()
87 ParseScalarArgument<tstring>(ctx, "field_delim", &delim)); in MakeDataset()
92 OP_REQUIRES_OK(ctx, ParseScalarArgument<bool>(ctx, "header", &header)); in MakeDataset()
95 OP_REQUIRES_OK(ctx, ParseScalarArgument<bool>(ctx, "use_quote_delim", in MakeDataset()
99 ParseScalarArgument<tstring>(ctx, "na_value", &na_value)); in MakeDataset()
Dset_stats_aggregator_dataset_op.cc92 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, "tag", &tag)); in MakeDataset()
94 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, "counter_prefix", &prefix)); in MakeDataset()
Dstats_dataset_ops.cc46 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, "tag", &tag)); in MakeDataset()
169 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, "tag", &tag)); in MakeDataset()
Dparallel_interleave_dataset_op.cc1192 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, kCycleLength, &cycle_length)); in MakeDataset()
1197 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, kBlockLength, &block_length)); in MakeDataset()
1203 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, kSloppy, &sloppy)); in MakeDataset()
1214 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, kBufferOutputElements, in MakeDataset()
1220 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, kPrefetchInputElements, in MakeDataset()
Dsleep_dataset_op.cc32 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, "sleep_microseconds", in MakeDataset()

12