Home
last modified time | relevance | path

Searched refs:_variant_tensor (Results 1 – 25 of 44) sorted by relevance

12

/external/tensorflow/tensorflow/python/data/ops/
Doptional_ops.py174 self._variant_tensor = variant_tensor
178 with ops.colocate_with(self._variant_tensor):
179 return gen_dataset_ops.optional_has_value(self._variant_tensor, name=name)
185 [self._variant_tensor]) as scope:
186 with ops.colocate_with(self._variant_tensor):
188 self._variant_tensor,
246 return [value._variant_tensor] # pylint: disable=protected-access
Ddataset_ops.py205 self._variant_tensor,
222 def _variant_tensor(self): member in DatasetV2
225 @_variant_tensor.setter
226 def _variant_tensor(self, _): member in DatasetV2
254 self._variant_tensor,
259 self._variant_tensor,
263 self._variant_tensor, allow_stateful=allow_stateful)
274 variant = self._variant_tensor
422 with ops.colocate_with(self._variant_tensor):
2214 dataset._variant_tensor,
[all …]
Dreaders.py182 variant_tensor = self._impl._variant_tensor # pylint: disable=protected-access
279 self._input_dataset._variant_tensor, # pylint: disable=protected-access
347 variant_tensor = self._impl._variant_tensor # pylint: disable=protected-access
503 variant_tensor = self._impl._variant_tensor # pylint: disable=protected-access
Dmulti_device_iterator_ops.py267 self._dataset._variant_tensor, # pylint: disable=protected-access
353 self._dataset._variant_tensor,
363 ds_variant = ds._variant_tensor
542 dataset._variant_tensor, # pylint: disable=protected-access
/external/tensorflow/tensorflow/python/data/kernel_tests/
Doptional_test.py188 [opt1._variant_tensor, opt2._variant_tensor])
196 [opt_none1._variant_tensor, opt_none2._variant_tensor])
209 opt3 = optional_ops.Optional.from_value((5.0, opt1._variant_tensor))
210 opt4 = optional_ops.Optional.from_value((6.0, opt2._variant_tensor))
213 [opt3._variant_tensor, opt4._variant_tensor])
230 zeros_tensor = array_ops.zeros_like(opt._variant_tensor)
236 zeros_tensor = array_ops.zeros_like(opt_none._variant_tensor)
249 opt2 = optional_ops.Optional.from_value(opt1._variant_tensor)
251 zeros_tensor = array_ops.zeros_like(opt2._variant_tensor)
271 array_ops.identity(optional_with_value._variant_tensor),
[all …]
Dplacement_test.py114 with ops.colocate_with(dataset._variant_tensor): # pylint:disable=protected-access
165 self.assertIn("gpu:0", dataset._variant_tensor.device.lower())
185 with ops.colocate_with(dataset._variant_tensor):
216 with ops.colocate_with(dataset._variant_tensor):
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dtesting.py94 self._input_dataset._variant_tensor, # pylint: disable=protected-access
108 self._input_dataset._variant_tensor, # pylint: disable=protected-access
120 self._input_dataset._variant_tensor, # pylint: disable=protected-access
Ddistribute.py74 self._input_dataset._variant_tensor, # pylint: disable=protected-access
148 input_dataset._variant_tensor, # pylint: disable=protected-access
287 input_dataset._variant_tensor, # pylint: disable=protected-access
325 dataset_device = dataset._variant_tensor.device
332 with ops.colocate_with(dataset._variant_tensor):
524 return ged_ops.compute_batch_size(dataset._variant_tensor) # pylint: disable=protected-access
Derror_ops.py69 self._input_dataset._variant_tensor, # pylint: disable=protected-access
75 self._input_dataset._variant_tensor, # pylint: disable=protected-access
Dcardinality.py68 …return gen_dataset_ops.dataset_cardinality(dataset._variant_tensor) # pylint: disable=protected-a…
113 self._input_dataset._variant_tensor, # pylint: disable=protected-access
Doptimization.py103 … [dataset._variant_tensor for dataset in self._datasets], # pylint: disable=protected-access
210 input_dataset._variant_tensor, # pylint: disable=protected-access
Dbatching.py315 self._input_dataset._variant_tensor, # pylint: disable=protected-access
362 self._input_dataset._variant_tensor, # pylint: disable=protected-access
447 variant = self._mapped_dataset._variant_tensor # pylint: disable=protected-access
Dscan_ops.py129 self._input_dataset._variant_tensor,
138 self._input_dataset._variant_tensor,
Dget_single_element.py69 dataset._variant_tensor, **dataset._flat_structure)) # pylint: disable=protected-access
Dsleep.py31 self._input_dataset._variant_tensor, # pylint: disable=protected-access
Dinterleave_ops.py138 self._selector_input._variant_tensor,
139 [data_input._variant_tensor for data_input in self._data_inputs],
Dunique.py63 self._input_dataset._variant_tensor, # pylint: disable=protected-access
Dsnapshot.py85 self._input_dataset._variant_tensor, # pylint: disable=protected-access
240 input_dataset._variant_tensor, # pylint: disable=protected-access
Dtake_while_ops.py45 self._input_dataset._variant_tensor, # pylint: disable=protected-access
Dwriters.py115 …dataset._variant_tensor, self._filename, self._compression_type) # pylint: disable=protected-acce…
Dstats_ops.py105 self._input_dataset._variant_tensor, # pylint: disable=protected-access
Dprefetching_ops.py105 self._input_dataset._variant_tensor) # pylint: disable=protected-access
242 self._input_dataset._variant_tensor, # pylint: disable=protected-access
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dwrap_unwrap_test.py36 ds_variant = ds._variant_tensor # pylint: disable=protected-access
50 ds_variant = ds._variant_tensor # pylint: disable=protected-access
/external/tensorflow/tensorflow/python/distribute/
Dinput_ops.py65 remap_dict = _clone_helper(dataset._variant_tensor.op, variant_tensor_ops)
66 new_variant_tensor = remap_dict[dataset._variant_tensor.op].outputs[0]
/external/tensorflow/tensorflow/python/data/util/
Dtraverse.py41 bfs_q.put(dataset._variant_tensor.op) # pylint: disable=protected-access

12