Home
last modified time | relevance | path

Searched refs:_make_dataset (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
Dmap_vectorization_test.py349 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/
Dtraining_generator_test.py339 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/
Ddataset_ops.py1441 def _make_dataset(): function
1455 _make_dataset.add_to_graph(ops.get_default_graph())
1470 dataset_factory=_make_dataset, **flat_structure(self)),