Searched refs:graph_rewrites (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | optimize_dataset_test.py | 386 graph_rewrites = options._graph_rewrites()
393 (", ".join(graph_rewrites.enabled + graph_rewrites.default)))
426 graph_rewrites = options._graph_rewrites()
427 self.assertEqual(set(graph_rewrites.enabled),
429 self.assertEqual(set(graph_rewrites.disabled),
431 self.assertEqual(set(graph_rewrites.default),
435 graph_rewrites = options._graph_rewrites()
436 self.assertEqual(set(graph_rewrites.enabled),
438 self.assertEqual(set(graph_rewrites.disabled),
440 self.assertEqual(set(graph_rewrites.default),
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | optimization_options.py | 57 graph_rewrites = options.graph_rewrites()
58 result = graph_rewrites(enabled=[], disabled=[], default=[])
|
| /external/tensorflow/tensorflow/python/data/util/ |
| D | options.py | 72 def graph_rewrites(): function
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | dataset_ops.py | 384 graph_rewrites = options._graph_rewrites()
388 if graph_rewrites.enabled or graph_rewrites.default:
394 ", ".join(graph_rewrites.enabled + graph_rewrites.default))
395 elif (graph_rewrites.enabled or graph_rewrites.default or
398 dataset = _OptimizeDataset(dataset, graph_rewrites.enabled,
399 graph_rewrites.disabled,
400 graph_rewrites.default, graph_rewrite_configs)
3112 graph_rewrites = options_lib.graph_rewrites()
3113 return graph_rewrites(enabled=list(set(result.enabled)),
|