Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/contrib/ignite/kernels/dataset/
Dignite_dataset_ops.cc102 ctx, ParseScalarArgument<string>(ctx, "cache_name", &cache_name)); in MakeDataset()
108 OP_REQUIRES_OK(ctx, ParseScalarArgument<string>(ctx, "host", &host)); in MakeDataset()
117 OP_REQUIRES_OK(ctx, ParseScalarArgument<int32>(ctx, "port", &port)); in MakeDataset()
123 OP_REQUIRES_OK(ctx, ParseScalarArgument<bool>(ctx, "local", &local)); in MakeDataset()
132 OP_REQUIRES_OK(ctx, ParseScalarArgument<int32>(ctx, "part", &part)); in MakeDataset()
143 ParseScalarArgument<int32>(ctx, "page_size", &page_size)); in MakeDataset()
/external/tensorflow/tensorflow/contrib/cloud/kernels/
Dgcs_config_ops.cc66 Status ParseScalarArgument(OpKernelContext* ctx, StringPiece argument_name, in ParseScalarArgument() function
87 OP_REQUIRES_OK(ctx, ParseScalarArgument<string>(ctx, "json", &json_string)); in Compute()
183 OP_REQUIRES_OK(ctx, ParseScalarArgument<size_t>(ctx, "max_cache_size", in Compute()
186 ParseScalarArgument<size_t>(ctx, "block_size", &block_size)); in Compute()
188 ctx, ParseScalarArgument<size_t>(ctx, "max_staleness", &max_staleness)); in Compute()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dto_tf_record_op.cc35 Status ParseScalarArgument(OpKernelContext* ctx, in ParseScalarArgument() function in tensorflow::data::__anon18e6484b0111::ToTFRecordOp
52 ctx, ParseScalarArgument<string>(ctx, "filename", &filename), done); in ComputeAsync()
55 ParseScalarArgument<string>(ctx, "compression_type", in ComputeAsync()
Dauto_shard_dataset_op.cc38 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, "num_workers", &num_workers)); in MakeDataset()
43 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, "index", &index)); in MakeDataset()
Dsql_dataset_op.cc57 ctx, ParseScalarArgument<string>(ctx, "driver_name", &driver_name)); in MakeDataset()
60 OP_REQUIRES_OK(ctx, ParseScalarArgument<string>(ctx, "data_source_name", in MakeDataset()
64 OP_REQUIRES_OK(ctx, ParseScalarArgument<string>(ctx, "query", &query)); in MakeDataset()
Drandom_dataset_op.cc36 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, "seed", &seed)); in MakeDataset()
39 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, "seed2", &seed2)); in MakeDataset()
Dsliding_window_dataset_op.cc40 ctx, ParseScalarArgument<int64>(ctx, "window_size", &window_size)); in MakeDataset()
46 ctx, ParseScalarArgument<int64>(ctx, "window_shift", &window_shift)); in MakeDataset()
52 ctx, ParseScalarArgument<int64>(ctx, "window_stride", &window_stride)); in MakeDataset()
Dcsv_dataset_op.cc43 OP_REQUIRES_OK(ctx, ParseScalarArgument<string>(ctx, "compression_type", in MakeDataset()
66 ctx, ParseScalarArgument<int64>(ctx, "buffer_size", &buffer_size)); in MakeDataset()
72 ParseScalarArgument<string>(ctx, "field_delim", &delim)); in MakeDataset()
77 OP_REQUIRES_OK(ctx, ParseScalarArgument<bool>(ctx, "header", &header)); in MakeDataset()
80 OP_REQUIRES_OK(ctx, ParseScalarArgument<bool>(ctx, "use_quote_delim", in MakeDataset()
84 ParseScalarArgument<string>(ctx, "na_value", &na_value)); in MakeDataset()
Dset_stats_aggregator_dataset_op.cc86 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, "tag", &tag)); in MakeDataset()
88 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, "counter_prefix", &prefix)); in MakeDataset()
Dstats_dataset_ops.cc45 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, "tag", &tag)); in MakeDataset()
155 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, "tag", &tag)); in MakeDataset()
Drebatch_dataset_op.cc37 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, "num_workers", &num_workers)); in MakeDataset()
Dparallel_interleave_dataset_op.cc50 ParseScalarArgument(ctx, "cycle_length", &cycle_length)); in MakeDataset()
56 ParseScalarArgument(ctx, "block_length", &block_length)); in MakeDataset()
61 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, "sloppy", &sloppy)); in MakeDataset()
64 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, "buffer_output_elements", in MakeDataset()
71 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, "prefetch_input_elements", in MakeDataset()
/external/tensorflow/tensorflow/core/kernels/data/
Drange_dataset_op.cc32 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, "start", &start)); in MakeDataset()
35 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, "stop", &stop)); in MakeDataset()
38 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, "step", &step)); in MakeDataset()
Dshuffle_dataset_op.cc357 ctx, ParseScalarArgument<int64>(ctx, "buffer_size", &buffer_size)); in MakeDataset()
363 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, "seed", &seed)); in MakeDataset()
366 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, "seed2", &seed2)); in MakeDataset()
630 ctx, ParseScalarArgument<int64>(ctx, "buffer_size", &buffer_size)); in MakeDataset()
636 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, "seed", &seed)); in MakeDataset()
639 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, "seed2", &seed2)); in MakeDataset()
642 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, "count", &count)); in MakeDataset()
Dreader_dataset_ops.cc47 OP_REQUIRES_OK(ctx, ParseScalarArgument<string>(ctx, "compression_type", in MakeDataset()
52 ctx, ParseScalarArgument<int64>(ctx, "buffer_size", &buffer_size)); in MakeDataset()
301 ctx, ParseScalarArgument<int64>(ctx, "header_bytes", &header_bytes)); in MakeDataset()
307 ctx, ParseScalarArgument<int64>(ctx, "record_bytes", &record_bytes)); in MakeDataset()
313 ctx, ParseScalarArgument<int64>(ctx, "footer_bytes", &footer_bytes)); in MakeDataset()
319 ctx, ParseScalarArgument<int64>(ctx, "buffer_size", &buffer_size)); in MakeDataset()
327 OP_REQUIRES_OK(ctx, ParseScalarArgument<string>(ctx, "compression_type", in MakeDataset()
758 OP_REQUIRES_OK(ctx, ParseScalarArgument<string>(ctx, "compression_type", in MakeDataset()
763 ctx, ParseScalarArgument<int64>(ctx, "buffer_size", &buffer_size)); in MakeDataset()
Dwindow_dataset_op.cc36 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, "size", &window_size)); in MakeDataset()
43 ParseScalarArgument<int64>(ctx, "shift", &window_shift)); in MakeDataset()
50 ParseScalarArgument<int64>(ctx, "stride", &window_stride)); in MakeDataset()
57 ctx, ParseScalarArgument<bool>(ctx, "drop_remainder", &drop_remainder)); in MakeDataset()
Dshard_dataset_op.cc38 ParseScalarArgument<int64>(ctx, "num_shards", &num_shards)); in MakeDataset()
45 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, "index", &index)); in MakeDataset()
Dbatch_dataset_op.cc37 ParseScalarArgument<int64>(ctx, "batch_size", &batch_size)); in MakeDataset()
44 OP_REQUIRES_OK(ctx, ParseScalarArgument<bool>(ctx, "drop_remainder", in MakeDataset()
/external/tensorflow/tensorflow/contrib/bigtable/kernels/
Dbigtable_scan_dataset_op.cc29 OP_REQUIRES_OK(ctx, ParseScalarArgument<string>(ctx, "prefix", &prefix)); in MakeDataset()
32 ParseScalarArgument<string>(ctx, "start_key", &start_key)); in MakeDataset()
34 OP_REQUIRES_OK(ctx, ParseScalarArgument<string>(ctx, "end_key", &end_key)); in MakeDataset()
61 ctx, ParseScalarArgument<float>(ctx, "probability", &probability)); in MakeDataset()
Dbigtable_sample_key_pairs_dataset_op.cc30 OP_REQUIRES_OK(ctx, ParseScalarArgument<string>(ctx, "prefix", &prefix)); in MakeDataset()
34 ParseScalarArgument<string>(ctx, "start_key", &start_key)); in MakeDataset()
36 OP_REQUIRES_OK(ctx, ParseScalarArgument<string>(ctx, "end_key", &end_key)); in MakeDataset()
Dbigtable_range_key_dataset_op.cc30 ParseScalarArgument<string>(ctx, "start_key", &start_key)); in MakeDataset()
32 OP_REQUIRES_OK(ctx, ParseScalarArgument<string>(ctx, "end_key", &end_key)); in MakeDataset()
Dbigtable_kernels.cc234 ctx, ParseScalarArgument<int64>(ctx, "timestamp", &timestamp_int), in ComputeAsync()
341 Status ParseScalarArgument(OpKernelContext* ctx, StringPiece argument_name, in ParseScalarArgument() function in tensorflow::data::__anon24f989ab0311::ToBigtableOp
Dbigtable_prefix_key_dataset_op.cc29 OP_REQUIRES_OK(ctx, ParseScalarArgument<string>(ctx, "prefix", &prefix)); in MakeDataset()
/external/tensorflow/tensorflow/contrib/kinesis/kernels/
Dkinesis_dataset_ops.cc152 ParseScalarArgument<std::string>(ctx, "stream", &stream)); in MakeDataset()
154 OP_REQUIRES_OK(ctx, ParseScalarArgument<std::string>(ctx, "shard", &shard)); in MakeDataset()
156 OP_REQUIRES_OK(ctx, ParseScalarArgument<bool>(ctx, "read_indefinitely", in MakeDataset()
159 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, "interval", &interval)); in MakeDataset()
/external/tensorflow/tensorflow/contrib/kafka/kernels/
Dkafka_dataset_ops.cc41 ParseScalarArgument<std::string>(ctx, "servers", &servers)); in MakeDataset()
43 OP_REQUIRES_OK(ctx, ParseScalarArgument<std::string>(ctx, "group", &group)); in MakeDataset()
45 OP_REQUIRES_OK(ctx, ParseScalarArgument<bool>(ctx, "eof", &eof)); in MakeDataset()
47 OP_REQUIRES_OK(ctx, ParseScalarArgument<int64>(ctx, "timeout", &timeout)); in MakeDataset()

12