Searched refs:_make_dataset (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| D | map_vectorization_test.py | 349 def _make_dataset(node_names): function
359 unoptimized = _make_dataset([map_node_name, "Batch"])
362 optimized = _make_dataset(["ChooseFastestBranch"]
419 def _make_dataset(node_names): function
424 unoptimized = _make_dataset(["MapAndBatch"])
425 optimized = _make_dataset(["ChooseFastestBranch"])
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_generator_test.py | 339 def _make_dataset(self, inputs, batches): member in TestConvertToGeneratorLike
344 self._make_dataset(inputs, batches))
358 ('simple_dataset', _make_dataset, simple_inputs),
362 ('nested_dataset', _make_dataset, nested_inputs),
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | dataset_ops.py | 1441 def _make_dataset(): function
1455 _make_dataset.add_to_graph(ops.get_default_graph())
1470 dataset_factory=_make_dataset, **flat_structure(self)),
|