/external/tensorflow/tensorflow/core/kernels/data/ |
D | random_seed_ops.cc | 71 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()
|
D | fixed_length_record_dataset_op.cc | 479 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()
|
D | shuffle_dataset_op.cc | 635 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()
|
D | range_dataset_op.cc | 306 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()
|
D | window_dataset_op.cc | 397 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()
|
D | shard_dataset_op.cc | 249 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, kNumShards, &num_shards)); in MakeDataset() 256 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, kIndex, &index)); in MakeDataset()
|
D | batch_dataset_op.cc | 267 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, kBatchSize, &batch_size)); in MakeDataset() 274 ctx, ParseScalarArgument<bool>(ctx, kDropRemainder, &drop_remainder)); in MakeDataset()
|
D | text_line_dataset_op.cc | 248 OP_REQUIRES_OK(ctx, ParseScalarArgument<tstring>(ctx, kCompressionType, in MakeDataset() 253 ParseScalarArgument<int64>(ctx, kBufferSize, &buffer_size)); in MakeDataset()
|
D | tf_record_dataset_op.cc | 299 OP_REQUIRES_OK(ctx, ParseScalarArgument<tstring>(ctx, kCompressionType, in MakeDataset() 304 ParseScalarArgument<int64>(ctx, kBufferSize, &buffer_size)); in MakeDataset()
|
D | parallel_batch_dataset_op.cc | 531 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()
|
D | interleave_dataset_op.cc | 323 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, kCycleLength, &cycle_length)); in MakeDataset() 332 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, kBlockLength, &block_length)); in MakeDataset()
|
D | dataset_ops.cc | 123 ParseScalarArgument(ctx, kGraphDef, &graph_def_string)); in Compute()
|
/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
D | to_tf_record_op.cc | 38 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()
|
D | data_service_ops.cc | 45 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, kAddress, &address)); in Compute() 50 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, kProtocol, &protocol)); in Compute()
|
D | auto_shard_dataset_op.cc | 48 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, kNumWorkers, &num_workers)); in MakeDataset() 53 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, kIndex, &index)); in MakeDataset()
|
D | sql_dataset_op.cc | 55 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()
|
D | sampling_dataset_op.cc | 231 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()
|
D | data_service_dataset_op.cc | 885 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()
|
D | random_dataset_op.cc | 154 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, "seed", &seed)); in MakeDataset() 157 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, "seed2", &seed2)); in MakeDataset()
|
D | sliding_window_dataset_op.cc | 38 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()
|
D | csv_dataset_op.cc | 46 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()
|
D | set_stats_aggregator_dataset_op.cc | 92 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, "tag", &tag)); in MakeDataset() 94 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, "counter_prefix", &prefix)); in MakeDataset()
|
D | stats_dataset_ops.cc | 46 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, "tag", &tag)); in MakeDataset() 169 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, "tag", &tag)); in MakeDataset()
|
D | parallel_interleave_dataset_op.cc | 1192 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()
|
D | sleep_dataset_op.cc | 32 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, "sleep_microseconds", in MakeDataset()
|