Home
last modified time | relevance | path

Searched refs:variant_tensor (Results 1 – 25 of 26) sorted by relevance

12

/external/tensorflow/tensorflow/python/data/experimental/ops/
Dtesting.py92 variant_tensor = (
97 super(_AssertNextDataset, self).__init__(input_dataset, variant_tensor)
106 variant_tensor = (
110 super(_NonSerializableDataset, self).__init__(input_dataset, variant_tensor)
119 variant_tensor = gen_experimental_dataset_ops.sleep_dataset(
123 super(_SleepDataset, self).__init__(input_dataset, variant_tensor)
Ddistribute.py78 variant_tensor = ged_ops.auto_shard_dataset(
86 super(_AutoShardDataset, self).__init__(input_dataset, variant_tensor)
152 variant_tensor = ged_ops.rebatch_dataset_v2(
157 super(_RebatchDataset, self).__init__(input_dataset, variant_tensor)
291 variant_tensor = ged_ops.rebatch_dataset(
295 super(_LegacyRebatchDataset, self).__init__(input_dataset, variant_tensor)
308 variant_tensor = ged_ops.dataset_from_graph(graph_def)
309 super(_RemoteDataset, self).__init__(variant_tensor)
Dsleep.py30 variant_tensor = gen_experimental_dataset_ops.sleep_dataset(
34 super(_SleepDataset, self).__init__(input_dataset, variant_tensor)
Derror_ops.py65 variant_tensor = (
70 super(_IgnoreErrorsDataset, self).__init__(input_dataset, variant_tensor)
Dio.py154 variant_tensor = gen_experimental_dataset_ops.save_dataset_v2(
164 super(_SaveDataset, self).__init__(dataset, variant_tensor)
231 variant_tensor = gen_experimental_dataset_ops.load_dataset(
237 super(_LoadDataset, self).__init__(variant_tensor)
Dmatching_files.py34 variant_tensor = ged_ops.matching_files_dataset(self._patterns)
35 super(MatchingFilesDataset, self).__init__(variant_tensor)
Dcardinality.py112 variant_tensor = ged_ops.assert_cardinality_dataset(
117 variant_tensor)
Dprefetching_ops.py209 variant_tensor = gen_dataset_ops.generator_dataset(
217 super(_CopyToDeviceDataset, self).__init__(input_dataset, variant_tensor)
246 variant_tensor = ged_ops.experimental_map_dataset(
252 super(_MapOnGpuDataset, self).__init__(input_dataset, variant_tensor)
Dbatching.py314 variant_tensor = ged_ops.dense_to_sparse_batch_dataset(
320 variant_tensor)
361 variant_tensor = ged_ops.map_and_batch_dataset(
370 super(_MapAndBatchDataset, self).__init__(input_dataset, variant_tensor)
Dshuffle_ops.py43 variant_tensor = gen_dataset_ops.shuffle_and_repeat_dataset(
51 variant_tensor)
Dsnapshot.py81 variant_tensor = ged_ops.snapshot_dataset(
100 super(_LegacySnapshotDataset, self).__init__(input_dataset, variant_tensor)
Dparsing_ops.py87 variant_tensor = (
101 super(_ParseExampleDataset, self).__init__(input_dataset, variant_tensor)
Dgrouping.py273 variant_tensor = ged_ops.experimental_group_by_reducer_dataset(
284 super(_GroupByReducerDataset, self).__init__(input_dataset, variant_tensor)
Dinterleave_ops.py136 variant_tensor = (
143 super(_DirectedInterleaveDataset, self).__init__(variant_tensor)
Dreaders.py783 variant_tensor = gen_experimental_dataset_ops.csv_dataset_v2(
795 super(CsvDatasetV2, self).__init__(variant_tensor)
1170 variant_tensor = gen_experimental_dataset_ops.sql_dataset(
1173 super(SqlDatasetV2, self).__init__(variant_tensor)
Ddata_service_ops.py287 variant_tensor = gen_experimental_dataset_ops.data_service_dataset_v2(
301 super(_DataServiceDatasetV2, self).__init__(variant_tensor)
/external/tensorflow/tensorflow/python/data/ops/
Dreaders.py132 variant_tensor = gen_dataset_ops.text_line_dataset(self._filenames,
135 super(_TextLineDataset, self).__init__(variant_tensor)
214 variant_tensor = self._impl._variant_tensor # pylint: disable=protected-access
216 super(TextLineDatasetV2, self).__init__(variant_tensor)
270 variant_tensor = gen_dataset_ops.tf_record_dataset(self._filenames,
273 super(_TFRecordDataset, self).__init__(variant_tensor)
310 variant_tensor = ged_ops.legacy_parallel_interleave_dataset_v2(
321 variant_tensor)
420 variant_tensor = self._impl._variant_tensor # pylint: disable=protected-access
421 super(TFRecordDatasetV2, self).__init__(variant_tensor)
[all …]
Ddataset_ops.py204 def __init__(self, variant_tensor): argument
214 self._variant_tensor_attr = variant_tensor
3056 variant_tensor = self._as_variant_tensor()
3069 super(DatasetV1, self).__init__(variant_tensor)
3758 def __init__(self, input_dataset, variant_tensor): argument
3760 super(UnaryDataset, self).__init__(variant_tensor)
3769 def __init__(self, input_dataset, variant_tensor): argument
3772 input_dataset, variant_tensor)
3788 variant_tensor = gen_dataset_ops.tensor_dataset(
3791 super(TensorDataset, self).__init__(variant_tensor)
[all …]
Doptional_ops.py173 def __init__(self, variant_tensor, element_spec): argument
174 self._variant_tensor = variant_tensor
Dmulti_device_iterator_ops.py133 variant_tensor = gen_dataset_ops.generator_dataset(
141 super(_PerDeviceGenerator, self).__init__(variant_tensor)
175 variant_tensor = gen_dataset_ops.generator_dataset(
183 super(_ReincarnatedPerDeviceGenerator, self).__init__(variant_tensor)
/external/tensorflow/tensorflow/python/data/benchmarks/
Dfrom_tensor_slices_benchmark.py42 variant_tensor = gen_dataset_ops.flat_map_dataset(
48 variant_tensor)
/external/tensorflow/tensorflow/python/data/util/
Dtraverse_test.py41 variant_tensor = gen_dataset_ops.model_dataset(
43 super(_TestDataset, self).__init__(input_dataset, variant_tensor)
/external/tensorflow/tensorflow/python/distribute/coordinator/
Dvalues.py399 variant_tensor = ged_ops.dataset_from_graph(graph_def)
400 super(_RemoteDataset, self).__init__(variant_tensor)
/external/tensorflow/tensorflow/python/distribute/
Dinput_ops_test.py269 variant_tensor = gen_dataset_ops.model_dataset(
271 super(_TestDataset, self).__init__(input_dataset, variant_tensor)
/external/tensorflow/tensorflow/python/kernel_tests/
Dconstant_op_test.py147 variant_tensor = tensor_pb2.TensorProto(
156 const = constant_op.constant(variant_tensor)
160 self.assertProtoEquals(variant_tensor, const_value)
573 variant_tensor = tensor_pb2.TensorProto(
582 const_variant = constant_op.constant(variant_tensor)

12