Home
last modified time | relevance | path

Searched refs:grappler_item (Results 1 – 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/python/grappler/
Ditem_test.py58 grappler_item = item.Item(mg)
59 op_list = grappler_item.IdentifyImportantOps()
71 grappler_item = item.Item(mg)
72 op_properties = grappler_item.GetOpProperties()
75 for node in grappler_item.metagraph.graph_def.node:
93 grappler_item = item.Item(mg)
95 initial_tf_item = grappler_item.tf_item
96 no_change_tf_item = grappler_item.tf_item
100 for node in grappler_item.metagraph.graph_def.node:
102 new_tf_item = grappler_item.tf_item
[all …]
Ddatasets_test.py56 grappler_item = item.Item(mg)
57 op_properties = grappler_item.GetOpProperties()
81 grappler_item = item.Item(mg)
82 op_properties = grappler_item.GetOpProperties()
117 grappler_item = item.Item(mg)
118 op_properties = grappler_item.GetOpProperties()
130 grappler_item = item.Item(mg)
131 op_properties = grappler_item.GetOpProperties()
157 grappler_item = item.Item(mg)
158 op_properties = grappler_item.GetOpProperties()
[all …]
Dcluster_test.py42 grappler_item = item.Item(mg)
46 grappler_item)
59 grappler_item = item.Item(mg)
63 grappler_item)
77 grappler_item = item.Item(mg)
80 peak_mem = grappler_cluster.DeterminePeakMemoryUsage(grappler_item)
97 grappler_item = item.Item(mg)
109 op_perfs, run_time, _ = grappler_cluster.MeasureCosts(grappler_item)
124 grappler_item = item.Item(mg)
128 op_perfs, run_time, step_stats = gcluster.MeasureCosts(grappler_item)
[all …]
Dtf_optimizer_wrapper.cc86 std::unique_ptr<tensorflow::grappler::GrapplerItem> grappler_item = in PYBIND11_MODULE() local
89 if (!grappler_item) { in PYBIND11_MODULE()
99 optimizer.Optimize(cluster, *grappler_item, &out_graph)); in PYBIND11_MODULE()
Ditem_wrapper.cc125 py::class_<tensorflow::grappler::GrapplerItem> grappler_item( in PYBIND11_MODULE() local
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
DBUILD52 "//tensorflow/core/grappler:grappler_item",
76 "//tensorflow/core/grappler:grappler_item",
97 "//tensorflow/core/grappler:grappler_item",
115 "//tensorflow/core/grappler:grappler_item",
135 "//tensorflow/core/grappler:grappler_item",
149 "//tensorflow/core/grappler:grappler_item",
170 "//tensorflow/core/grappler:grappler_item",
184 "//tensorflow/core/grappler:grappler_item",
205 "//tensorflow/core/grappler:grappler_item",
219 "//tensorflow/core/grappler:grappler_item",
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/
DBUILD42 "//tensorflow/core/grappler:grappler_item",
62 "//tensorflow/core/grappler:grappler_item",
81 "//tensorflow/core/grappler:grappler_item",
99 "//tensorflow/core/grappler:grappler_item",
118 "//tensorflow/core/grappler:grappler_item",
154 "//tensorflow/core/grappler:grappler_item",
176 "//tensorflow/core/grappler:grappler_item",
205 "//tensorflow/core/grappler:grappler_item",
234 "//tensorflow/core/grappler:grappler_item",
252 "//tensorflow/core/grappler:grappler_item",
[all …]
/external/tensorflow/tensorflow/core/grappler/
DBUILD13 "grappler_item.h",
57 ":grappler_item",
138 ":grappler_item",
151 name = "grappler_item",
153 "grappler_item.cc",
155 hdrs = ["grappler_item.h"],
176 ":grappler_item",
195 ":grappler_item",
250 ":grappler_item",
/external/tensorflow/tensorflow/core/grappler/costs/
DBUILD84 "//tensorflow/core/grappler:grappler_item",
112 "//tensorflow/core/grappler:grappler_item",
129 "//tensorflow/core/grappler:grappler_item",
145 "//tensorflow/core/grappler:grappler_item",
279 "//tensorflow/core/grappler:grappler_item",
318 "//tensorflow/core/grappler:grappler_item",
373 "//tensorflow/core/grappler:grappler_item",
/external/tensorflow/tensorflow/core/grappler/optimizers/inference/
DBUILD30 "//tensorflow/core/grappler:grappler_item",
51 "//tensorflow/core/grappler:grappler_item",
/external/tensorflow/tensorflow/core/grappler/inputs/
DBUILD61 "//tensorflow/core/grappler:grappler_item",
77 "//tensorflow/core/grappler:grappler_item",
/external/tensorflow/tensorflow/core/grappler/clusters/
DBUILD81 "//tensorflow/core/grappler:grappler_item",
114 "//tensorflow/core/grappler:grappler_item",
162 "//tensorflow/core/grappler:grappler_item",
/external/tensorflow/tensorflow/core/kernels/data/
Drewrite_utils.cc119 std::unique_ptr<tensorflow::grappler::GrapplerItem> grappler_item = in ApplyRewrites() local
124 grappler_item->optimization_options().optimize_function_library = false; in ApplyRewrites()
132 std::move(*grappler_item), config, ctx->device(), &cluster, graph_def)); in ApplyRewrites()
DBUILD128 "//tensorflow/core/grappler:grappler_item",
997 "//tensorflow/core/grappler:grappler_item",
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.h34 const grappler::GrapplerItem* grappler_item = nullptr; member
Dconvert_graph.cc698 GraphDef modified_graph_def = params.grappler_item->graph; in ConvertAfterShapes()
715 grappler::GrapplerItem grappler_item = in ConvertAfterShapes() local
716 params.grappler_item->WithGraph(std::move(modified_graph_def)); in ConvertAfterShapes()
717 const GraphDef& graph_def = grappler_item.graph; in ConvertAfterShapes()
719 grappler::GraphProperties static_graph_properties(grappler_item); in ConvertAfterShapes()
Dconvert_graph_test.cc170 params.grappler_item = &item; in RunConvertAfterShape()
Dtrt_optimization_pass.cc232 cp.grappler_item = &item; in Optimize()
/external/tensorflow/tensorflow/core/grappler/verifiers/
DBUILD60 "//tensorflow/core/grappler:grappler_item",
/external/tensorflow/tensorflow/core/grappler/utils/
DBUILD50 "//tensorflow/core/grappler:grappler_item",
162 "//tensorflow/core/grappler:grappler_item",
199 "//tensorflow/core/grappler:grappler_item",
/external/tensorflow/tensorflow/compiler/tf2tensorrt/
DBUILD409 "//tensorflow/core/grappler:grappler_item",
448 "//tensorflow/core/grappler:grappler_item",
/external/tensorflow/tensorflow/tools/def_file_filter/
Dsymbols_pybind.txt264 [//tensorflow/core/grappler:grappler_item] # tf_item
/external/tensorflow/tensorflow/core/common_runtime/eager/
DBUILD434 "//tensorflow/core/grappler:grappler_item",
/external/tensorflow/tensorflow/core/common_runtime/
DBUILD1634 "//tensorflow/core/grappler:grappler_item",
1643 "//tensorflow/core/grappler:grappler_item",
1665 "//tensorflow/core/grappler:grappler_item",
/external/tensorflow/tensorflow/python/
DBUILD289 "//tensorflow/core/grappler:grappler_item",
339 "//tensorflow/core/grappler:grappler_item",
5252 "//tensorflow/core/grappler:grappler_item",
5327 "//tensorflow/core/grappler:grappler_item", # tf_item

12