Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/python/data/ops/
Diterator_ops.py75 output_shapes, output_classes): argument
98 or output_classes is None):
103 output_types, output_shapes, output_classes)
114 output_classes=None): argument
187 if output_classes is None:
188 output_classes = nest.map_structure(lambda _: ops.Tensor, output_types)
191 output_types, output_shapes, output_classes)
217 output_classes)
223 output_classes=None): argument
275 if output_classes is None:
[all …]
Ddataset_ops.py1291 output_classes = wrapped_func.output_classes
1294 nest.flatten(output_classes), nest.flatten(state_classes)):
1299 wrapped_func.output_classes))
1531 def output_classes(self): member in DatasetV1
1567 self.output_types, self.output_shapes, self.output_classes)
2416 def output_classes(self): member in StructuredFunctionWrapper
2561 output_classes = get_legacy_output_classes(input_dataset)
2562 if output_classes != get_legacy_output_classes(dataset_to_concatenate):
2565 (output_classes, get_legacy_output_classes(dataset_to_concatenate)))
2577 output_types, output_shapes, output_classes)
/external/tensorflow/tensorflow/contrib/eager/python/examples/densenet/
Ddensenet_test.py36 output_classes = 10
43 output_classes, num_layers_in_each_block,
53 self.assertEqual(output_shape, (batch_size, output_classes))
59 output_classes = 10
66 output_classes, num_layers_in_each_block,
76 self.assertEqual(output_shape, (batch_size, output_classes))
82 output_classes = 10
89 output_classes, num_layers_in_each_block,
99 self.assertEqual(output_shape, (batch_size, output_classes))
189 self.output_classes = 1000
[all …]
Ddensenet_graph_test.py54 output_classes = 10
60 output_classes, num_layers_in_each_block,
72 self.assertAllEqual([batch_size, output_classes], out.shape)
81 self.output_classes = 1000
96 self.output_classes,
126 self.output_classes,
Ddensenet.py174 output_classes, num_layers_in_each_block, data_format, argument
181 self.output_classes = output_classes
246 self.classifier = tf.keras.layers.Dense(self.output_classes)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.data.-iterator.pbtxt11 name: "output_classes"
24 …_resource\', \'initializer\', \'output_types\', \'output_shapes\', \'output_classes\'], varargs=No…
28 …argspec: "args=[\'string_handle\', \'output_types\', \'output_shapes\', \'output_classes\'], varar…
32 …argspec: "args=[\'output_types\', \'output_shapes\', \'shared_name\', \'output_classes\'], varargs…
Dtensorflow.data.-dataset.pbtxt7 name: "output_classes"
Dtensorflow.data.-t-f-record-dataset.pbtxt9 name: "output_classes"
Dtensorflow.data.experimental.-csv-dataset.pbtxt9 name: "output_classes"
Dtensorflow.data.-text-line-dataset.pbtxt9 name: "output_classes"
Dtensorflow.data.experimental.-sql-dataset.pbtxt9 name: "output_classes"
Dtensorflow.data.-fixed-length-record-dataset.pbtxt9 name: "output_classes"
Dtensorflow.data.experimental.-random-dataset.pbtxt9 name: "output_classes"
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dscan_ops.py70 new_state_classes, self._output_classes = wrapped_func.output_classes
73 new_state_classes, output_classes = wrapped_func.output_classes
99 output_types, output_shapes, output_classes)
Dindexed_dataset_ops.py35 def __init__(self, materialized_resource, materializer, output_classes, argument
39 self._output_classes = output_classes
103 self.output_classes, self.output_types,
Dparsing_ops.py73 output_classes = dict(
79 output_types, output_shapes, output_classes)
Dgrouping.py295 state_classes = self._init_func.output_classes
308 nest.flatten(wrapped_func.output_classes),
314 (self._state_classes, wrapped_func.output_classes))
Dbatching.py485 output_classes=None, argument
550 if output_classes is None:
552 output_classes = nest.pack_sequence_as(
556 output_types, output_shapes, output_classes)
Doptimization.py255 input_dataset.output_classes)))
/external/tensorflow/tensorflow/python/data/util/
Dstructure.py244 def convert_legacy_structure(output_types, output_shapes, output_classes): argument
273 flat_classes = nest.flatten(output_classes)
290 ret = nest.pack_sequence_as(output_classes, flat_ret)
Dstructure_test.py357 output_classes, expected_structure): argument
359 output_types, output_shapes, output_classes)
/external/tensorflow/tensorflow/python/distribute/
Dinput_lib.py215 def output_classes(self): member in InputIteratorImpl
216 return self._iterators[0].output_classes
456 def output_classes(self): member in _SingleWorkerDatasetIterator
/external/tensorflow/tensorflow/contrib/data/python/ops/
Dbatching.py226 output_classes=dataset_ops.get_legacy_output_classes(dataset))
/external/tensorflow/tensorflow/contrib/ignite/python/ops/
Dignite_dataset_ops.py249 output_classes = {}
251 output_classes[field.name] = field.to_output_classes()
252 return output_classes
/external/tensorflow/tensorflow/contrib/legacy_seq2seq/python/kernel_tests/
Dseq2seq_test.py752 output_classes = 5
755 i + 0.5, shape=[2, output_classes]) for i in range(3)

12