HomeSort by relevance Sort by last modified time
    Searched refs:input_dataset (Results 1 - 14 of 14) sorted by null

  /external/tensorflow/tensorflow/contrib/data/python/ops/
grouping.py 118 def __init__(self, input_dataset, key_func, reduce_func, window_size_func):
122 self._input_dataset = input_dataset
124 self._make_key_func(key_func, input_dataset)
125 self._make_reduce_func(reduce_func, input_dataset)
144 def _make_key_func(self, key_func, input_dataset):
148 sparse.as_dense_types(input_dataset.output_types,
149 input_dataset.output_classes)))
152 # Pass in shape information from the input_dataset.
153 dense_shapes = sparse.as_dense_shapes(input_dataset.output_shapes,
154 input_dataset.output_classes
    [all...]
unique.py 54 def __init__(self, input_dataset):
57 self._input_dataset = input_dataset
58 if input_dataset.output_types not in (dtypes.int32, dtypes.int64,
interleave_ops.py 34 def __init__(self, input_dataset, map_func, cycle_length, block_length,
38 self._input_dataset = input_dataset
41 sparse.as_dense_types(input_dataset.output_types,
42 input_dataset.output_classes)))
45 # Pass in shape information from the input_dataset.
46 dense_shapes = sparse.as_dense_shapes(input_dataset.output_shapes,
47 input_dataset.output_classes)
51 nested_args = nest.pack_sequence_as(input_dataset.output_types, args)
53 nested_args, input_dataset.output_types, input_dataset.output_shapes
    [all...]
error_ops.py 59 def __init__(self, input_dataset):
62 self._input_dataset = input_dataset
scan_ops.py 33 def __init__(self, input_dataset, initial_state, scan_func):
36 self._input_dataset = input_dataset
72 sparse.as_dense_types(input_dataset.output_types,
73 input_dataset.output_classes))))
76 # Pass in shape information from the state and input_dataset.
78 dense_shapes = sparse.as_dense_shapes(input_dataset.output_shapes,
79 input_dataset.output_classes)
86 input_value = nest.pack_sequence_as(input_dataset.output_types,
batching.py 226 def __init__(self, input_dataset, batch_size, row_shape):
229 if not isinstance(input_dataset.output_types, dtypes.DType):
232 input_dataset.output_types)
233 self._input_dataset = input_dataset
351 def __init__(self, input_dataset, map_func, batch_size, num_parallel_batches):
353 super(_MapAndBatchDataset, self).__init__(input_dataset, map_func)
stats_ops.py 155 def __init__(self, input_dataset, op_function, tag):
157 self._input_dataset = input_dataset
shuffle_ops.py 34 input_dataset,
40 self._input_dataset = input_dataset
  /external/tensorflow/tensorflow/contrib/training/python/training/
tensor_queue_dataset.py 39 def __init__(self, input_dataset, batch_size, padded_shapes, padding_values):
42 if sparse.any_sparse(input_dataset.output_classes):
45 self._input_dataset = input_dataset
51 dataset_ops._partial_shape_to_tensor, input_dataset.output_shapes)
54 input_dataset.output_shapes, dataset_ops._partial_shape_to_tensor,
58 dataset_ops._default_padding(input_dataset))
60 input_dataset.output_shapes, dataset_ops._padding_value_to_tensor,
61 padding_values, input_dataset.output_types)
  /external/tensorflow/tensorflow/python/data/ops/
dataset_ops.py     [all...]
  /external/tensorflow/tensorflow/python/data/kernel_tests/
concatenate_dataset_op_test.py 41 input_dataset = dataset_ops.Dataset.from_tensor_slices(input_components)
44 concatenated = input_dataset.concatenate(dataset_to_concatenate)
74 input_dataset = dataset_ops.Dataset.from_tensor_slices(input_components)
77 concatenated = input_dataset.concatenate(dataset_to_concatenate)
109 input_dataset = dataset_ops.Dataset.from_tensor_slices(input_components)
115 input_dataset.concatenate(dataset_to_concatenate)
125 input_dataset = dataset_ops.Dataset.from_tensor_slices(input_components)
130 input_dataset.concatenate(dataset_to_concatenate)
batch_dataset_op_test.py 338 input_dataset = dataset_ops.Dataset.from_tensors(
342 dynamic_padding_from_tensor_shapes = input_dataset.padded_batch(
347 dynamic_padding_from_lists = input_dataset.padded_batch(
349 dynamic_padding_from_lists_with_minus_one = input_dataset.padded_batch(
351 dynamic_padding_from_tensors = input_dataset.padded_batch(
  /external/tensorflow/tensorflow/contrib/data/python/kernel_tests/
bucketing_test.py 215 input_dataset = (
218 bucketed_dataset = input_dataset.apply(
250 input_dataset = (
253 bucketed_dataset = input_dataset.apply(
320 input_dataset = (
324 bucketed_dataset = input_dataset.apply(
  /external/tensorflow/tensorflow/go/op/
wrappers.go     [all...]

Completed in 467 milliseconds