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)),
|