Home
last modified time | relevance | path

Searched refs:from_tensors (Results 1 – 25 of 87) sorted by relevance

1234

/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Ddirected_interleave_dataset_test.py37 dataset_ops.Dataset.from_tensors(i).repeat(100) for i in range(10)
63 dataset_ops.Dataset.from_tensors(i).repeat(None)
92 probs_ds = dataset_ops.Dataset.from_tensors(probs).repeat()
98 datasets = [dataset_ops.Dataset.from_tensors(w).repeat() for w in words]
124 dataset_ops.Dataset.from_tensors(0),
125 dataset_ops.Dataset.from_tensors(0.0)
130 dataset_ops.Dataset.from_tensors(0),
131 dataset_ops.Dataset.from_tensors(1)
132 ], choice_dataset=dataset_ops.Dataset.from_tensors(1.0))
136 dataset_ops.Dataset.from_tensors(0),
[all …]
Dunbatch_test.py43 dataset = dataset_ops.Dataset.from_tensors([0, 1, 2,
76 data = dataset_ops.Dataset.from_tensors(st)
90 data = dataset_ops.Dataset.from_tensors((list(range(10)), st))
126 data = dataset_ops.Dataset.from_tensors(
133 data = dataset_ops.Dataset.from_tensors((np.arange(7), np.arange(8),
143 data = dataset_ops.Dataset.from_tensors((ph1, ph2))
Dcardinality_test.py72 lambda _: dataset_ops.Dataset.from_tensors(0)), cardinality.UNKNOWN),
75 ("FromTensors1", lambda: dataset_ops.Dataset.from_tensors(0), 1),
76 ("FromTensors2", lambda: dataset_ops.Dataset.from_tensors((0, 1)), 1),
83 lambda _: dataset_ops.Dataset.from_tensors(0), cycle_length=1),
86 lambda _: dataset_ops.Dataset.from_tensors(0),
Dscan_test.py41 return dataset_ops.Dataset.from_tensors(0).repeat().apply(
62 data = dataset_ops.Dataset.from_tensors(1).repeat(None).apply(
109 dataset = dataset_ops.Dataset.from_tensors(0).repeat(5).apply(
161 dataset = dataset_ops.Dataset.from_tensors(0).apply(
Ddense_to_sparse_batch_test.py86 dataset_ops.Dataset.from_tensors(input_tensor).apply(
92 return dataset_ops.Dataset.from_tensors(input_tensor).apply(
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
Doptimize_dataset_test.py51 return dataset_ops.Dataset.from_tensors(0).map(lambda x: x + var)
54 return dataset_ops.Dataset.from_tensors(
55 0).flat_map(lambda _: dataset_ops.Dataset.from_tensors(var))
58 return dataset_ops.Dataset.from_tensors(0).filter(lambda x: x < var)
65 return dataset_ops.Dataset.from_tensors(0).apply(
80 return dataset_ops.Dataset.from_tensors(var)
82 return dataset_ops.Dataset.from_tensors(0).repeat(10).apply(
86 return dataset_ops.Dataset.from_tensors(0).apply(
118 dataset = dataset_ops.Dataset.from_tensors(input_t)
148 dataset = dataset_ops.Dataset.from_tensors(0)
[all …]
Dassert_next_dataset_test.py32 dataset = dataset_ops.Dataset.from_tensors(0).apply(
40 dataset = dataset_ops.Dataset.from_tensors(0).apply(
53 dataset = dataset_ops.Dataset.from_tensors(0).apply(
Dchoose_fastest_branch_dataset_test.py88 dataset = dataset_ops.Dataset.from_tensors(0).repeat(10)
120 dataset = dataset_ops.Dataset.from_tensors(0).repeat(1000).batch(100)
134 dataset = dataset_ops.Dataset.from_tensors(0).repeat(1000).batch(100)
158 dataset = dataset_ops.Dataset.from_tensors(0)
Dmap_vectorization_test.py187 return dataset_ops.Dataset.from_tensors((x_int, y_int))
190 return dataset_ops.Dataset.from_tensors((x > 0, y > 0))
193 return dataset_ops.Dataset.from_tensors((x, y))
268 return dataset_ops.Dataset.from_tensors(np.random.rand(10, 3)).repeat(5)
292 ("Cycle", map_fn_with_cycle, lambda: dataset_ops.Dataset.from_tensors(1)),
533 base_dataset = dataset_ops.Dataset.from_tensors(tensor)
545 base_dataset = dataset_ops.Dataset.from_tensors(0)
/external/tensorflow/tensorflow/contrib/distribute/python/examples/
Dsimple_estimator_example.py73 features = tf.data.Dataset.from_tensors([[1.]]).repeat(10)
74 labels = tf.data.Dataset.from_tensors([1.]).repeat(10)
82 features = tf.data.Dataset.from_tensors([[1.]]).repeat(10)
83 labels = tf.data.Dataset.from_tensors([1.]).repeat(10)
91 predict_features = tf.data.Dataset.from_tensors([[1.]]).repeat(10)
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dfrom_tensors_test.py46 dataset = dataset_ops.Dataset.from_tensors(components)
65 dataset = dataset_ops.Dataset.from_tensors(components)
84 dataset = dataset_ops.Dataset.from_tensors(components)
98 dataset = dataset_ops.Dataset.from_tensors(components)
141 lambda x, y: dataset_ops.Dataset.from_tensors(((x[0], x[1]),
181 dataset = dataset_ops.Dataset.from_tensors(components)
185 lambda x, y: dataset_ops.Dataset.from_tensors(
212 dataset = dataset_ops.Dataset.from_tensors(components)
229 dataset = dataset_ops.Dataset.from_tensors(components)
261 dataset = dataset_ops.Dataset.from_tensors(0)
Drepeat_test.py39 dataset = dataset_ops.Dataset.from_tensors(components).repeat(count)
57 dataset = dataset_ops.Dataset.from_tensors(components).repeat(-1)
72 dataset = dataset_ops.Dataset.from_tensors(components).repeat(
82 dataset = dataset_ops.Dataset.from_tensors(0).repeat(10).skip(10).repeat(-1)
Dmap_test.py363 dataset = dataset_ops.Dataset.from_tensors(0).repeat(
387 dataset = dataset_ops.Dataset.from_tensors(0).repeat(
400 dataset = dataset_ops.Dataset.from_tensors(0).repeat(
421 dataset = dataset_ops.Dataset.from_tensors(0).repeat(10).map(
499 dataset = dataset_ops.Dataset.from_tensors(
563 dataset = dataset_ops.Dataset.from_tensors(
605 dataset = dataset_ops.Dataset.from_tensors(
618 dataset = dataset_ops.Dataset.from_tensors(
856 dataset = dataset_ops.Dataset.from_tensors([1.0, 2.0, 3.0]).map(
863 dataset = dataset_ops.Dataset.from_tensors([1.0, 2.0, 3.0])
[all …]
Dflat_map_test.py43 lambda x: dataset_ops.Dataset.from_tensors([x]).repeat(x))
54 lambda y: dataset_ops.Dataset.from_tensors(y).repeat(y))
71 lambda y: dataset_ops.Dataset.from_tensors(y).repeat(y))),
99 lambda d: dataset_ops.Dataset.from_tensors(
Diterator_test.py171 dataset = dataset_ops.Dataset.from_tensors([1, 2, 3]).map(lambda x: x * x)
211 dataset = dataset_ops.Dataset.from_tensors(
253 dataset_ops.Dataset.from_tensors(components)
293 dataset_ops.Dataset.from_tensors(components))
303 dataset_3 = dataset_ops.Dataset.from_tensors(
305 dataset_4 = dataset_ops.Dataset.from_tensors(
382 dataset_ops.Dataset.from_tensors(((constant_op.constant(
389 dataset_ops.Dataset.from_tensors(
398 dataset_ops.Dataset.from_tensors(
559 dataset_float_vector = (dataset_ops.Dataset.from_tensors([1.0, 2.0, 3.0]))
[all …]
Dpadded_batch_test.py165 input_dataset = dataset_ops.Dataset.from_tensors(
211 _ = dataset_ops.Dataset.from_tensors([1, 2, 3]).padded_batch(
217 _ = dataset_ops.Dataset.from_tensors([1, 2, 3]).padded_batch(
223 _ = dataset_ops.Dataset.from_tensors([1, 2, 3]).padded_batch(
Dinterleave_test.py147 lambda x: dataset_ops.Dataset.from_tensors(x).repeat(x),
174 dataset_ops.Dataset.from_tensors, cycle_length, block_length,
227 lambda x: dataset_ops.Dataset.from_tensors(x).repeat(x),
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_utils_test.py108 lambda _: dataset_ops.Dataset.from_tensors(0)), ValueError),
112 ('FromTensors', lambda: dataset_ops.Dataset.from_tensors(0)),
116 lambda _: dataset_ops.Dataset.from_tensors(0), cycle_length=1),
119 lambda _: dataset_ops.Dataset.from_tensors(0),
156 lambda _: dataset_ops.Dataset.from_tensors(0)), ValueError),
160 ('FromTensors', lambda: dataset_ops.Dataset.from_tensors(0)),
164 lambda _: dataset_ops.Dataset.from_tensors(0), cycle_length=1),
172 lambda _: dataset_ops.Dataset.from_tensors(0),
/external/tensorflow/tensorflow/python/data/experimental/benchmarks/
Dautotune_benchmark.py42 dataset = dataset_ops.Dataset.from_tensors((np.random.rand(1, 4 * k),
78 dataset = dataset_ops.Dataset.from_tensors((np.random.rand(1, 4 * k),
116 dataset = dataset_ops.Dataset.from_tensors((np.random.rand(1, 4 * k),
157 dataset_a = dataset_ops.Dataset.from_tensors(a).repeat()
158 dataset_b = dataset_ops.Dataset.from_tensors(b).repeat()
159 dataset_c = dataset_ops.Dataset.from_tensors(c).repeat()
Doptimize_benchmark.py45 dataset = dataset_ops.Dataset.from_tensors(0).repeat(None)
86 dataset = dataset_ops.Dataset.from_tensors(0).repeat(None)
127 dataset = dataset_ops.Dataset.from_tensors(5).repeat(None)
Dunbatch_benchmark.py40 dataset = dataset_ops.Dataset.from_tensors("element").repeat(None)
77 dataset = dataset_ops.Dataset.from_tensors("element").repeat(None)
/external/tensorflow/tensorflow/python/grappler/
Ddatasets_test.py50 dataset = dataset_ops.Dataset.from_tensors(test_case['tensor'])
149 dataset = dataset_ops.Dataset.from_tensors(test_case['tensor'])
165 return dataset.concatenate(dataset_ops.Dataset.from_tensors(tensor))
253 dataset = dataset_ops.Dataset.from_tensors(test_case['tensor'])
282 dataset = dataset_ops.Dataset.from_tensors(test_case['tensor'])
316 return dataset_ops.Dataset.from_tensors(tensor).repeat(n)
350 return dataset_ops.Dataset.from_tensors(tensor).repeat(n)
383 dataset = dataset_ops.Dataset.from_tensors(test_case['tensor'])
/external/tensorflow/tensorflow/python/data/benchmarks/
Dmap_benchmark.py31 dataset = dataset_ops.Dataset.from_tensors(0).repeat(None)
50 dataset = dataset_ops.Dataset.from_tensors(
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dresampling.py68 initial_dist_ds = dataset_ops.Dataset.from_tensors(
70 acceptance_dist_ds = dataset_ops.Dataset.from_tensors(
72 prob_of_original_ds = dataset_ops.Dataset.from_tensors(
/external/tensorflow/tensorflow/contrib/distribute/python/
Dsingle_loss_example.py36 return dataset_ops.Dataset.from_tensors([[1.]]).repeat()
60 dataset = dataset_ops.Dataset.from_tensors([[1.]]).repeat()

1234