Home
last modified time | relevance | path

Searched refs:optimized_graph_def (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/tools/
Doptimize_for_inference_test.py162 optimized_graph_def = optimize_for_inference_lib.fold_batch_norms(
167 optimized_graph_def, input_map={}, name="optimized")
172 for node in optimized_graph_def.node:
214 optimized_graph_def = optimize_for_inference_lib.fold_batch_norms(
218 optimized_graph_def, input_map={}, name="optimized")
224 for node in optimized_graph_def.node:
244 optimized_graph_def = optimize_for_inference_lib.fuse_resize_and_conv(
249 optimized_graph_def, input_map={}, name="optimized")
254 for node in optimized_graph_def.node:
274 optimized_graph_def = optimize_for_inference_lib.fuse_resize_and_conv(
[all …]
Doptimize_for_inference_lib.py108 optimized_graph_def = input_graph_def
109 optimized_graph_def = strip_unused_lib.strip_unused(
110 optimized_graph_def, input_node_names, output_node_names,
112 optimized_graph_def = graph_util.remove_training_nodes(
113 optimized_graph_def, output_node_names)
114 optimized_graph_def = fold_batch_norms(optimized_graph_def)
116 optimized_graph_def = fuse_resize_and_conv(optimized_graph_def,
118 ensure_graph_is_valid(optimized_graph_def)
119 return optimized_graph_def
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_control_flow_test.cc98 GraphDef optimized_graph_def; in TEST() local
99 graph.ToGraphDef(&optimized_graph_def); in TEST()
101 FunctionalizeControlFlowForGraphDef(&optimized_graph_def, &library)); in TEST()
106 for (const GraphDef& graph_def : {optimized_graph_def, converted_graph_def}) { in TEST()
252 GraphDef optimized_graph_def; in TEST() local
253 graph.ToGraphDef(&optimized_graph_def); in TEST()
255 FunctionalizeControlFlowForGraphDef(&optimized_graph_def, &library)); in TEST()
260 for (const GraphDef& graph_def : {optimized_graph_def, converted_graph_def}) { in TEST()
400 GraphDef optimized_graph_def; in TEST() local
401 graph.ToGraphDef(&optimized_graph_def); in TEST()
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/
Ddependency_optimizer_test.cc729 GraphDef optimized_graph_def; in TEST_F() local
730 Status status = optimizer.Optimize(nullptr, item, &optimized_graph_def); in TEST_F()
733 EXPECT_EQ(6, optimized_graph_def.node_size()); in TEST_F()