| /external/tensorflow/tensorflow/core/kernels/data/ |
| parallel_interleave_dataset_op.cc | 21 #include "tensorflow/core/kernels/data/dataset.h"
87 new Dataset(input, std::move(captured_func), cycle_length, block_length,
93 class Dataset : public DatasetBase {
95 Dataset(const DatasetBase* input,
112 ~Dataset() override { input_->Unref(); }
128 return "ParallelInterleaveDatasetOp::Dataset";
154 // function must be a dataset. The worker thread then repeatedly calls
184 class Iterator : public DatasetIterator<Dataset> {
187 : DatasetIterator<Dataset>(params),
188 input_impl_(params.dataset->input_->MakeIterator(params.prefix))
[all...] |
| tensor_slice_dataset_op.cc | 18 #include "tensorflow/core/kernels/data/dataset.h"
33 // Create a new TensorDatasetOp::Dataset, insert it in the step
53 *output = new Dataset(ctx, std::move(components));
57 class Dataset : public GraphDatasetBase {
59 explicit Dataset(OpKernelContext* ctx, std::vector<Tensor> tensors)
84 string DebugString() override { return "TensorSliceDatasetOp::Dataset"; }
104 class Iterator : public DatasetIterator<Dataset> {
107 : DatasetIterator<Dataset>(params),
109 n_(params.dataset->tensors_[0].dim_size(0)) {}
117 out_tensors->reserve(dataset()->tensors_.size())
[all...] |
| scan_dataset_op.cc | 22 #include "tensorflow/core/kernels/data/dataset.h"
66 *output = new Dataset(ctx, input, func_, std::move(initial_state),
72 class Dataset : public GraphDatasetBase {
74 Dataset(OpKernelContext* ctx, const DatasetBase* input,
91 ~Dataset() override { input_->Unref(); }
106 string DebugString() override { return "ScanDatasetOp::Dataset"; }
148 class Iterator : public DatasetIterator<Dataset> {
151 : DatasetIterator<Dataset>(params),
152 input_impl_(params.dataset->input_->MakeIterator(params.prefix)),
153 state_(params.dataset->initial_state_) {
[all...] |
| random_dataset_op.cc | 18 #include "tensorflow/core/kernels/data/dataset.h"
48 *output = new Dataset(ctx, seed, seed2);
52 class Dataset : public GraphDatasetBase {
54 Dataset(OpKernelContext* ctx, int64 seed, int64 seed2)
76 ")::Dataset");
91 class Iterator : public DatasetIterator<Dataset> {
94 : DatasetIterator<Dataset>(params),
95 parent_generator_(dataset()->seed_, dataset()->seed2_),
123 random::PhiloxRandom(dataset()->seed_, dataset()->seed2_)
[all...] |
| sql_dataset_ops.cc | 19 #include "tensorflow/core/kernels/data/dataset.h"
73 *output = new Dataset(driver_name, data_source_name, query, output_types_,
78 class Dataset : public DatasetBase {
80 Dataset(const string& driver_name, const string& data_source_name,
103 string DebugString() override { return "SqlDatasetOp::Dataset"; }
106 class Iterator : public DatasetIterator<Dataset> {
109 : DatasetIterator<Dataset>(params) {}
126 dataset()->driver_name_);
127 Status s = query_connection_->Open(dataset()->data_source_name_,
128 dataset()->query_
[all...] |
| zip_dataset_op.cc | 17 #include "tensorflow/core/kernels/data/dataset.h"
37 *output = new Dataset(ctx, inputs);
41 class Dataset : public GraphDatasetBase {
43 explicit Dataset(OpKernelContext* ctx,
57 ~Dataset() override {
77 string DebugString() override { return "ZipDatasetOp::Dataset"; }
95 class Iterator : public DatasetIterator<Dataset> {
98 : DatasetIterator<Dataset>(params) {
99 input_impls_.reserve(params.dataset->inputs_.size());
101 for (const auto& input : params.dataset->inputs_)
[all...] |
| shuffle_dataset_op.cc | 21 #include "tensorflow/core/kernels/data/dataset.h"
41 // Abstract base dataset that implements a shuffling iterator.
75 buffer_.reset(new std::vector<Tensor>[params.dataset->buffer_size_]);
88 input_impl_ = dataset()->input_->MakeIterator(prefix());
90 while (input_impl_ && num_elements_ < dataset()->buffer_size_) {
95 << num_elements_ << " of " << dataset()->buffer_size_;
99 while (dataset()->count_ == -1 || epoch_ < dataset()->count_) {
106 if (first_call && dataset()->count_ == -1) {
117 input_impl_ = dataset()->input_->MakeIterator(prefix())
[all...] |
| parallel_map_dataset_op.cc | 21 #include "tensorflow/core/kernels/data/dataset.h"
64 *output = new Dataset(ctx, input, func_, num_parallel_calls, output_types_,
69 class Dataset : public GraphDatasetBase {
71 Dataset(OpKernelContext* ctx, const DatasetBase* input,
86 ~Dataset() override { input_->Unref(); }
102 string DebugString() override { return "ParallelMapDatasetOp::Dataset"; }
149 class Iterator : public DatasetIterator<Dataset> {
152 : DatasetIterator<Dataset>(params),
153 input_impl_(params.dataset->input_->MakeIterator(params.prefix)),
154 invocation_results_(params.dataset->num_parallel_calls_) {
[all...] |
| interleave_dataset_op.cc | 20 #include "tensorflow/core/kernels/data/dataset.h"
74 new Dataset(ctx, input, func_, std::move(captured_func), cycle_length,
79 class Dataset : public GraphDatasetBase {
81 Dataset(OpKernelContext* ctx, const DatasetBase* input,
97 ~Dataset() override { input_->Unref(); }
112 string DebugString() override { return "InterleaveDatasetOp::Dataset"; }
148 class Iterator : public DatasetIterator<Dataset> {
151 : DatasetIterator<Dataset>(params),
152 input_impl_(params.dataset->input_->MakeIterator(params.prefix)),
153 current_elements_(params.dataset->cycle_length_)
[all...] |
| repeat_dataset_op.cc | 17 #include "tensorflow/core/kernels/data/dataset.h"
34 // Create a new RepeatDatasetOp::Dataset, insert it in the step-local
38 *output = new Dataset(ctx, count, input);
42 class Dataset : public GraphDatasetBase {
44 Dataset(OpKernelContext* ctx, int64 count, const DatasetBase* input)
49 ~Dataset() override { input_->Unref(); }
72 string DebugString() override { return "RepeatDatasetOp::Dataset"; }
87 class EmptyIterator : public DatasetIterator<Dataset> {
90 : DatasetIterator<Dataset>(params) {}
108 class FiniteIterator : public DatasetIterator<Dataset> {
[all...] |
| /external/tensorflow/tensorflow/contrib/kafka/kernels/ |
| kafka_dataset_ops.cc | 16 #include "tensorflow/core/kernels/dataset.h"
53 *output = new Dataset(ctx, std::move(topics), servers, group, eof, timeout);
57 class Dataset : public GraphDatasetBase {
59 Dataset(OpKernelContext* ctx, std::vector<string> topics,
86 string DebugString() override { return "KafkaDatasetOp::Dataset"; }
107 class Iterator : public DatasetIterator<Dataset> {
110 : DatasetIterator<Dataset>(params) {}
126 consumer_->consume(dataset()->timeout_));
141 dataset()->eof_) {
160 if (current_topic_index_ == dataset()->topics_.size())
[all...] |
| /external/tensorflow/tensorflow/contrib/data/python/kernel_tests/ |
| interleave_dataset_op_test.py | 46 return dataset_ops.Dataset.from_tensor_slices(input_values).repeat(
48 lambda x: dataset_ops.Dataset.from_tensors(x).repeat(x),
87 return dataset_ops.Dataset.from_tensor_slices(
91 return dataset_ops.Dataset.range(10).map(_map_fn).interleave(
135 dataset = dataset_ops.Dataset.from_tensors(x)
136 dataset = dataset.repeat(x)
137 return dataset.map(map_fn)
139 self.dataset =
[all...] |
| get_single_element_test.py | 36 dataset = (dataset_ops.Dataset.range(100)
41 element = get_single_element.get_single_element(dataset)
49 "Dataset was empty."):
53 "Dataset had more than one element."):
|
| /cts/tests/autofillservice/src/android/autofillservice/cts/ |
| AuthenticationActivity.java | 45 * This class simulates authentication at the dataset at reponse level
90 * Creates an {@link IntentSender} with the given unique id for the given dataset.
93 CannedDataset dataset) {
94 return createSender(context, id, dataset, null);
98 CannedDataset dataset, Bundle outClientState) {
101 sDatasets.put(id, dataset);
217 final CannedDataset dataset = local
229 } else if (dataset != null) {
230 result = dataset.asDataset((id) -> Helper.findNodeByResourceId(structure, id));
232 throw new IllegalStateException("no dataset or response")
[all...] |
| /external/tensorflow/tensorflow/contrib/eager/python/examples/rnn_ptb/ |
| rnn_ptb_graph_test.py | 81 dataset = tf.data.Dataset.from_tensors(
85 inputs = dataset.make_one_shot_iterator().get_next()
121 dataset = tf.data.Dataset.from_tensors(
126 dataset = tf.data.Dataset.zip((dataset, dataset))
127 (inputs, labels) = dataset.make_one_shot_iterator().get_next(
[all...] |
| /developers/build/prebuilts/gradle/AutofillFramework/afservice/src/main/java/com/example/android/autofill/service/ |
| AutofillHelper.java | 20 import android.service.autofill.Dataset;
45 * Wraps autofill data in a LoginCredential Dataset object which can then be sent back to the
48 public static Dataset newDataset(Context context,
53 Dataset.Builder datasetBuilder;
55 datasetBuilder = new Dataset.Builder
62 datasetBuilder = new Dataset.Builder
98 Dataset dataset = newDataset(context, autofillFields, local
100 if (dataset != null) {
101 responseBuilder.addDataset(dataset);
[all...] |
| /external/tensorflow/tensorflow/contrib/eager/python/examples/linear_regression/ |
| linear_regression.py | 70 def fit(model, dataset, optimizer, verbose=False, logdir=None):
75 dataset: The tf.data.Dataset to use for training data.
93 for i, (xs, ys) in enumerate(tfe.Iterator(dataset)):
107 """tf.data.Dataset that yields synthetic data for linear regression."""
126 return tf.data.Dataset.range(num_batches).map(batch)
144 dataset = synthetic_dataset(true_w, true_b, noise_level, batch_size, 20)
150 fit(model, dataset, optimizer, verbose=True, logdir=FLAGS.logdir)
|
| linear_regression_graph_test.py | 32 dataset = linear_regression.synthetic_dataset_helper(
39 iterator = dataset.make_initializable_iterator()
|
| /external/tensorflow/tensorflow/contrib/eager/python/ |
| datasets_test.py | 25 from tensorflow.python.data import Dataset
40 for t in datasets.Iterator(Dataset.range(4)):
45 iterator = datasets.Iterator(Dataset.range(4))
54 ds = Dataset.range(4)
64 ds = Dataset.zip((Dataset.range(4), Dataset.zip((Dataset.range(4),
65 Dataset.range(4)))))
80 it = datasets.Iterator(Dataset.range(8).map(math_ops.square).filter(even)
[all...] |
| datasets.py | 49 """An iterator producing tf.Tensor objects from a tf.data.Dataset."""
51 def __init__(self, dataset):
52 """Creates a new iterator over the given dataset.
56 dataset = tf.data.Dataset.range(4)
57 for x in Iterator(dataset):
65 dataset: A `tf.data.Dataset` object.
74 "tf.data.Dataset.make_iterator or "
75 "tf.data.Dataset.make_one_shot_iterator for graph construction"
[all...] |
| /developers/build/prebuilts/gradle/RecyclerView/kotlinApp/app/src/main/java/com/example/android/recyclerview/ |
| RecyclerViewFragment.kt | 38 private lateinit var dataset: Array<String>
45 // Initialize dataset, this data would usually come from a local content provider or
73 recyclerView.adapter = CustomAdapter(dataset)
130 dataset = Array(DATASET_COUNT, {i -> "This is element # $i"})
|
| /developers/samples/android/input/autofill/AutofillFramework/afservice/src/main/java/com/example/android/autofill/service/data/ |
| ClientAutofillDataBuilder.java | 59 "dataset-" + datasetNumber + "." + partition, mPackageName);
60 DatasetWithFilledAutofillFields dataset = local
62 if (dataset != null && dataset.filledAutofillFields != null) {
63 listBuilder.add(dataset);
70 * Parses a client view structure and build a dataset (in the form of a
73 private DatasetWithFilledAutofillFields buildDatasetForPartition(AutofillDataset dataset,
77 datasetWithFilledAutofillFields.autofillDataset = dataset;
|
| /developers/samples/android/ui/views/RecyclerView/kotlinApp/app/src/main/java/com/example/android/recyclerview/ |
| RecyclerViewFragment.kt | 38 private lateinit var dataset: Array<String>
45 // Initialize dataset, this data would usually come from a local content provider or
73 recyclerView.adapter = CustomAdapter(dataset)
130 dataset = Array(DATASET_COUNT, {i -> "This is element # $i"})
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| map_dataset_op_test.py | 51 return (dataset_ops.Dataset.from_tensor_slices(components).map(_map_fn)
55 """Test an dataset that maps a TF function across its input elements."""
63 dataset = self._buildMapDataset(components, count)
64 iterator = dataset.make_initializable_iterator()
112 return (dataset_ops.Dataset.from_tensor_slices(components)
118 """Test an dataset that maps a TF function across its input elements."""
128 dataset = self._buildParallelMapDataset(
130 iterator = dataset.make_initializable_iterator()
187 # Tests whether a parallel map dataset will be cleaned up correctly when
195 dataset = self._buildParallelMapDataset(components, 1000, 100, 100
[all...] |
| /external/autotest/tko/ |
| plotgraph.py | 49 title: title of the dataset
54 raise "plotgraph:add_dataset - dataset was empty! %s" %\
96 for dataset in self.datasets:
99 if label in dataset:
100 data = dataset[label]
|
