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 | 228 def _make_dataset(node_names): function
238 unoptimized = _make_dataset([map_node_name, "Batch"])
241 optimized = _make_dataset(["ChooseFastestBranch"]
503 def _make_dataset(node_names): function
508 unoptimized = _make_dataset(["MapAndBatch"])
509 optimized = _make_dataset(["ChooseFastestBranch"])
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_generator_test.py | 494 def _make_dataset(self, inputs, batches): member in TestConvertToGeneratorLike
499 self._make_dataset(inputs, batches))
513 ('simple_dataset', _make_dataset, simple_inputs),
517 ('nested_dataset', _make_dataset, nested_inputs),
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | dataset_ops.py | 2396 def _make_dataset(): function
2410 _make_dataset.add_to_graph(ops.get_default_graph())
2426 dataset_factory=_make_dataset, **self._flat_structure), None,
|