Home
last modified time | relevance | path

Searched refs:GrapplerItem (Results 1 – 25 of 156) sorted by relevance

1234567

/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item.h36 struct GrapplerItem { struct
37 GrapplerItem() = default;
38 GrapplerItem(const GrapplerItem& other) = default;
39 GrapplerItem(GrapplerItem&& other) = default;
40 GrapplerItem& operator=(const GrapplerItem& other) = default; argument
41 GrapplerItem& operator=(GrapplerItem&& other) = default; argument
42 virtual ~GrapplerItem() = default;
45 GrapplerItem WithGraph(GraphDef&& graph) const;
109 Status AddDevices(const GrapplerItem& other);
Dgrappler_item.cc33 GrapplerItem GrapplerItem::WithGraph(GraphDef&& graph_def) const { in WithGraph()
34 GrapplerItem item; in WithGraph()
51 std::vector<const NodeDef*> GrapplerItem::MainOpsFanin() const { in MainOpsFanin()
55 std::vector<const NodeDef*> GrapplerItem::EnqueueOpsFanin() const { in EnqueueOpsFanin()
65 std::vector<const NodeDef*> GrapplerItem::InitOpsFanin() const { in InitOpsFanin()
69 std::vector<const NodeDef*> GrapplerItem::MainVariables() const { in MainVariables()
80 std::unordered_set<string> GrapplerItem::NodesToPreserve() const { in NodesToPreserve()
132 const std::unordered_set<string>& GrapplerItem::devices() const { in devices()
136 Status GrapplerItem::AddDevice(const string& device) { in AddDevice()
152 Status GrapplerItem::AddDevices(const GrapplerItem& other) { in AddDevices()
[all …]
Dgrappler_item_builder_test.cc81 std::unique_ptr<GrapplerItem> item = in TEST_F()
163 std::unique_ptr<GrapplerItem> item = in TEST_F()
187 std::unique_ptr<GrapplerItem> item = in TEST_F()
207 std::unique_ptr<GrapplerItem> item = in TEST_F()
240 std::unique_ptr<GrapplerItem> item = in TEST_F()
272 std::unique_ptr<GrapplerItem> item = in TEST_F()
301 std::unique_ptr<GrapplerItem> item1 = in TEST_F()
308 std::unique_ptr<GrapplerItem> item2 = in TEST_F()
329 std::unique_ptr<GrapplerItem> item = in TEST_F()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmeta_optimizer_test.cc55 Status Optimize(Cluster* cluster, const GrapplerItem& item, in Optimize()
62 void Feedback(Cluster* cluster, const GrapplerItem& item, in Feedback()
95 gtl::FlatMap<string, GrapplerItem::OptimizationOptions>* in SetOptimizationOptions()
111 Status Optimize(Cluster* cluster, const GrapplerItem& item, in Optimize()
120 void Feedback(Cluster* cluster, const GrapplerItem& item, in Feedback()
124 static gtl::FlatMap<string, GrapplerItem::OptimizationOptions>*
128 gtl::FlatMap<string, GrapplerItem::OptimizationOptions>*
137 GrapplerItem item; in TEST_F()
156 GrapplerItem item; in TEST_F()
177 GrapplerItem item; in TEST_F()
[all …]
Dmeta_optimizer.h42 Status Optimize(Cluster* cluster, const GrapplerItem& item,
47 void Feedback(Cluster* cluster, const GrapplerItem& item,
76 Status OptimizeGraph(Cluster* cluster, const GrapplerItem& item,
95 GrapplerItem* optimized_item, GraphDef* optimized_graph,
109 Status RunMetaOptimizer(const GrapplerItem& item, const ConfigProto& cfg,
136 const GrapplerItem::OptimizationOptions& optimization_options,
Dlayout_optimizer_test.cc228 GrapplerItem item; in TEST_F()
263 GrapplerItem item; in TEST_F()
285 GrapplerItem item; in TEST_F()
298 GrapplerItem item; in TEST_F()
311 GrapplerItem item; in TEST_F()
324 GrapplerItem item; in TEST_F()
337 GrapplerItem item; in TEST_F()
350 GrapplerItem item; in TEST_F()
365 GrapplerItem item; in TEST_F()
392 GrapplerItem item; in TEST_F()
[all …]
Dfunction_optimizer_test.cc49 GrapplerItem item; in TEST_F()
113 GrapplerItem optimized = item.WithGraph(std::move(output)); in TEST_F()
143 GrapplerItem item; in TEST_F()
189 GrapplerItem optimized = item.WithGraph(std::move(output)); in TEST_F()
223 GrapplerItem item; in TEST_F()
289 GrapplerItem optimized = item.WithGraph(std::move(output)); in TEST_F()
314 GrapplerItem item; in TEST_F()
373 GrapplerItem optimized = item.WithGraph(std::move(output)); in TEST_F()
397 GrapplerItem item; in TEST_F()
423 GrapplerItem optimized = item.WithGraph(std::move(output)); in TEST_F()
[all …]
Dauto_parallel.h36 Status Optimize(Cluster* cluster, const GrapplerItem& item,
39 void Feedback(Cluster* cluster, const GrapplerItem& item,
48 const GrapplerItem* item_;
51 Status Initialize(const GrapplerItem& item);
Dmemory_optimizer_test.cc52 GrapplerItem item; in TEST_F()
91 GrapplerItem item; in TEST_F()
115 GrapplerItem item; in TEST_F()
155 GrapplerItem item; in TEST_F()
241 GrapplerItem item; in TEST_F()
282 GrapplerItem item_copy = item.WithGraph(std::move(output)); in TEST_F()
290 GrapplerItem optimized = item.WithGraph(std::move(output)); in TEST_F()
315 GrapplerItem item; in TEST_F()
340 GrapplerItem optimized = item.WithGraph(std::move(output)); in TEST_F()
366 GrapplerItem item; in TEST_F()
[all …]
Dimplementation_selector_test.cc45 GrapplerItem item; in TEST_F()
74 GrapplerItem item; in TEST_F()
112 GrapplerItem item; in TEST_F()
129 GrapplerItem optimized = item.WithGraph(std::move(output)); in TEST_F()
207 GrapplerItem item; in TEST_F()
225 GrapplerItem optimized = item.WithGraph(std::move(output)); in TEST_F()
Dlayout_optimizer.h44 Status Optimize(Cluster* cluster, const GrapplerItem& item,
47 void Feedback(Cluster* cluster, const GrapplerItem& item,
53 Status Tune(const GrapplerItem& item, const GraphProperties& graph_properties,
Dgraph_optimizer.h28 struct GrapplerItem;
43 virtual Status Optimize(Cluster* cluster, const GrapplerItem& item,
49 virtual void Feedback(Cluster* cluster, const GrapplerItem& item,
Dfunction_optimizer.h35 Status Optimize(Cluster* cluster, const GrapplerItem& item,
38 void Feedback(Cluster* cluster, const GrapplerItem& item,
57 const GrapplerItem& item, const GraphDef& graph, const int iteration,
Ddependency_optimizer_test.cc54 GrapplerItem item; in TEST_F()
77 GrapplerItem item; in TEST_F()
112 GrapplerItem item; in TEST_F()
155 GrapplerItem item; in TEST_F()
211 GrapplerItem item; in TEST_F()
249 GrapplerItem item; in TEST_F()
268 GrapplerItem item; in TEST_F()
311 GrapplerItem item; in TEST_F()
343 GrapplerItem item; in TEST_F()
367 GrapplerItem item; in TEST_F()
[all …]
Darithmetic_optimizer_test.cc93 GrapplerItem item; in TEST_F()
108 GrapplerItem item; in TEST_F()
145 GrapplerItem item; in TEST_F()
181 GrapplerItem item; in TEST_F()
220 GrapplerItem item; in TEST_F()
257 GrapplerItem item; in TEST_F()
290 GrapplerItem item; in TEST_F()
335 GrapplerItem item; in TEST_F()
361 GrapplerItem item; in TEST_F()
405 GrapplerItem item; in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties_test.cc125 GrapplerItem item; in TEST_F()
169 GrapplerItem item; in TEST_F()
196 GrapplerItem item; in TEST_F()
241 GrapplerItem item; in TEST_F()
290 GrapplerItem item; in TEST_F()
320 GrapplerItem item; in TEST_F()
360 GrapplerItem item; in TEST_F()
388 GrapplerItem item; in TEST_F()
406 GrapplerItem item; in TEST_F()
424 GrapplerItem item; in TEST_F()
[all …]
Danalytical_cost_estimator.h34 struct GrapplerItem;
55 Status Initialize(const GrapplerItem& item) override;
67 GrapplerItem item_;
Dgraph_memory_test.cc47 GrapplerItem item; in TEST_F()
75 GrapplerItem item; in TEST_F()
102 GrapplerItem item; in TEST_F()
139 GrapplerItem item; in TEST_F()
205 GrapplerItem item; in TEST_F()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmeta_optimizer.cc32 Status TFDataMetaOptimizer::Optimize(Cluster* cluster, const GrapplerItem& item, in Optimize()
35 GrapplerItem optimized_item = item; in Optimize()
67 GrapplerItem* item) const { in ApplyOptimization()
116 void TFDataMetaOptimizer::Feedback(Cluster* cluster, const GrapplerItem& item, in Feedback()
Dmeta_optimizer.h37 Status Optimize(Cluster* cluster, const GrapplerItem& item,
40 void Feedback(Cluster* cluster, const GrapplerItem& item,
50 GrapplerItem* item) const;
/external/tensorflow/tensorflow/python/grappler/
Dmodel_analyzer.h26 struct GrapplerItem;
33 explicit ModelAnalyzer(const GrapplerItem& item);
40 const GrapplerItem& item_;
Dcost_analyzer.h34 struct GrapplerItem;
52 explicit CostAnalyzer(const GrapplerItem& item, Cluster* cluster,
66 const GrapplerItem* item_;
/external/tensorflow/tensorflow/core/grappler/clusters/
Dsingle_machine_test.cc70 GrapplerItem item; in TEST_F()
103 GrapplerItem item; in TEST_F()
116 GrapplerItem item; in TEST_F()
177 GrapplerItem item; in TEST_F()
228 GrapplerItem item; in TEST_F()
242 GrapplerItem item; in RunInfiniteTFLoop()
359 GrapplerItem item; in TEST_F()
397 GrapplerItem item; in TEST_F()
490 GrapplerItem CreateGrapplerItemWithResourceMemory() { in CreateGrapplerItemWithResourceMemory()
524 GrapplerItem item; in CreateGrapplerItemWithResourceMemory()
[all …]
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dtrt_optimization_pass.h54 const grappler::GrapplerItem& item,
57 void Feedback(grappler::Cluster* cluster, const grappler::GrapplerItem& item,
61 const grappler::GrapplerItem& item);
/external/tensorflow/tensorflow/compiler/jit/
Dxla_fusion_optimizer_test.cc61 grappler::GrapplerItem item; in TEST_F()
93 grappler::GrapplerItem item; in TEST_F()
126 grappler::GrapplerItem item; in TEST_F()
150 grappler::GrapplerItem item; in TEST_F()
172 grappler::GrapplerItem item; in TEST_F()
198 grappler::GrapplerItem item; in TEST_F()

1234567