Home
last modified time | relevance | path

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

12345678

/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item.h37 struct GrapplerItem { struct
38 GrapplerItem() = default;
39 GrapplerItem(const GrapplerItem& other) = default;
40 GrapplerItem(GrapplerItem&& other) = default;
41 GrapplerItem& operator=(const GrapplerItem& other) = default; argument
42 GrapplerItem& operator=(GrapplerItem&& other) = default; argument
43 virtual ~GrapplerItem() = default;
46 GrapplerItem WithGraph(GraphDef&& graph) const;
113 Status AddDevices(const GrapplerItem& other);
136 GrapplerItem::OptimizationOptions CreateOptOptionsForEager(); argument
Dgrappler_item.cc34 GrapplerItem::OptimizationOptions CreateOptOptionsForEager() { in CreateOptOptionsForEager()
35 GrapplerItem::OptimizationOptions optimization_options; in CreateOptOptionsForEager()
52 GrapplerItem GrapplerItem::WithGraph(GraphDef&& graph_def) const { in WithGraph()
53 GrapplerItem item; in WithGraph()
70 std::vector<const NodeDef*> GrapplerItem::MainOpsFanin() const { in MainOpsFanin()
76 std::vector<const NodeDef*> GrapplerItem::EnqueueOpsFanin() const { in EnqueueOpsFanin()
88 std::vector<const NodeDef*> GrapplerItem::InitOpsFanin() const { in InitOpsFanin()
94 std::vector<const NodeDef*> GrapplerItem::MainVariables() const { in MainVariables()
106 std::unordered_set<string> GrapplerItem::NodesToPreserve() const { in NodesToPreserve()
170 const std::unordered_set<string>& GrapplerItem::devices() const { in devices()
[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()
365 std::unique_ptr<GrapplerItem> item = in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmeta_optimizer.h44 Status Optimize(Cluster* cluster, const GrapplerItem& item, in Optimize()
46 GrapplerItem copy(item); in Optimize()
50 Status OptimizeConsumeItem(Cluster* cluster, GrapplerItem&& item,
57 void Feedback(Cluster* cluster, const GrapplerItem& item, in Feedback()
89 Status OptimizeGraph(Cluster* cluster, GrapplerItem&& item,
109 GrapplerItem* optimized_item, GraphDef* optimized_graph,
123 Status RunMetaOptimizer(GrapplerItem&& item, const ConfigProto& cfg,
150 const GrapplerItem::OptimizationOptions& optimization_options,
Dmeta_optimizer_test.cc59 Status Optimize(Cluster* cluster, const GrapplerItem& item, in Optimize()
66 void Feedback(Cluster* cluster, const GrapplerItem& item, in Feedback()
99 gtl::FlatMap<string, GrapplerItem::OptimizationOptions>* in SetOptimizationOptions()
116 Status Optimize(Cluster* cluster, const GrapplerItem& item, in Optimize()
125 void Feedback(Cluster* cluster, const GrapplerItem& item, in Feedback()
129 static gtl::FlatMap<string, GrapplerItem::OptimizationOptions>*
133 gtl::FlatMap<string, GrapplerItem::OptimizationOptions>*
142 GrapplerItem item; in TEST_F()
161 GrapplerItem item; in TEST_F()
182 GrapplerItem item; in TEST_F()
[all …]
Dfunction_optimizer_test.cc44 GrapplerItem item; in TEST_F()
79 GrapplerItem optimized = item.WithGraph(std::move(output)); in TEST_F()
113 GrapplerItem item; in TEST_F()
138 GrapplerItem optimized = item.WithGraph(std::move(output)); in TEST_F()
163 GrapplerItem item; in TEST_F()
187 GrapplerItem optimized = item.WithGraph(std::move(output)); in TEST_F()
211 GrapplerItem item; in TEST_F()
244 GrapplerItem optimized = item.WithGraph(std::move(output)); in TEST_F()
270 GrapplerItem item; in TEST_F()
291 GrapplerItem optimized = item.WithGraph(std::move(output)); in TEST_F()
[all …]
Dgraph_optimizer.h31 struct GrapplerItem;
58 virtual Status Optimize(Cluster* cluster, const GrapplerItem& item,
62 virtual Status Optimize(Cluster* cluster, GrapplerItem&& item, in Optimize()
70 virtual void Feedback(Cluster* cluster, const GrapplerItem& item,
Dlayout_optimizer_test.cc234 GrapplerItem item; in TEST_F()
269 GrapplerItem item; in TEST_F()
291 GrapplerItem item; in TEST_F()
304 GrapplerItem item; in TEST_F()
317 GrapplerItem item; in TEST_F()
330 GrapplerItem item; in TEST_F()
343 GrapplerItem item; in TEST_F()
356 GrapplerItem item; in TEST_F()
369 GrapplerItem item; in TEST_F()
385 GrapplerItem item; in TEST_F()
[all …]
Dauto_parallel.h38 Status Optimize(Cluster* cluster, const GrapplerItem& item,
41 void Feedback(Cluster* cluster, const GrapplerItem& item,
50 const GrapplerItem* item_;
53 Status Initialize(const GrapplerItem& item);
Dimplementation_selector_test.cc47 GrapplerItem item; in TEST_F()
66 GrapplerItem item; in TEST_F()
90 GrapplerItem item; in TEST_F()
114 GrapplerItem item; in TEST_F()
146 GrapplerItem item; in TEST_F()
170 GrapplerItem item; in TEST_F()
193 GrapplerItem item; in TEST_F()
251 GrapplerItem item; in TEST_F()
317 GrapplerItem item; in TEST_F()
355 GrapplerItem item; in TEST_F()
[all …]
Dlayout_optimizer.h46 Status Optimize(Cluster* cluster, const GrapplerItem& item,
49 void Feedback(Cluster* cluster, const GrapplerItem& item,
55 Status Tune(const GrapplerItem& item, const GraphProperties& graph_properties,
Dfunction_optimizer.h38 Status Optimize(Cluster* cluster, const GrapplerItem& item,
41 void Feedback(Cluster* cluster, const GrapplerItem& item,
52 Status RunFunctionOptimizerPass(const GrapplerItem& item,
Dcommon_subgraph_elimination.h34 struct GrapplerItem;
49 Status Optimize(Cluster* cluster, const GrapplerItem& item,
52 void Feedback(Cluster* cluster, 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()
283 GrapplerItem item_copy = item.WithGraph(std::move(output)); in TEST_F()
291 GrapplerItem optimized = item.WithGraph(std::move(output)); in TEST_F()
316 GrapplerItem item; in TEST_F()
341 GrapplerItem optimized = item.WithGraph(std::move(output)); in TEST_F()
367 GrapplerItem item; in TEST_F()
[all …]
Ddependency_optimizer_test.cc56 GrapplerItem item; in TEST_F()
79 GrapplerItem item; in TEST_F()
114 GrapplerItem item; in TEST_F()
157 GrapplerItem item; in TEST_F()
212 GrapplerItem item; in TEST_F()
250 GrapplerItem item; in TEST_F()
269 GrapplerItem item; in TEST_F()
312 GrapplerItem item; in TEST_F()
344 GrapplerItem item; in TEST_F()
368 GrapplerItem item; in TEST_F()
[all …]
/external/tensorflow/tensorflow/python/grappler/
Dmodel_analyzer.h26 struct GrapplerItem;
33 explicit ModelAnalyzer(const GrapplerItem& item);
40 const GrapplerItem& item_;
Ditem_wrapper.cc122 PYBIND11_MAKE_OPAQUE(tensorflow::grappler::GrapplerItem);
125 py::class_<tensorflow::grappler::GrapplerItem> grappler_item( in PYBIND11_MODULE()
130 bool ignore_user_placement) -> tensorflow::grappler::GrapplerItem* { in PYBIND11_MODULE()
145 std::unique_ptr<tensorflow::grappler::GrapplerItem> item = in PYBIND11_MODULE()
156 [](tensorflow::grappler::GrapplerItem* item, in PYBIND11_MODULE()
192 [](tensorflow::grappler::GrapplerItem* item) in PYBIND11_MODULE()
214 [](tensorflow::grappler::GrapplerItem* item) in PYBIND11_MODULE()
Dcost_analyzer.h34 struct GrapplerItem;
52 explicit CostAnalyzer(const GrapplerItem& item, Cluster* cluster,
66 const GrapplerItem* item_;
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties_test.cc156 GrapplerItem item; in TEST_F()
197 GrapplerItem item; in TEST_F()
224 GrapplerItem item; in TEST_F()
279 GrapplerItem item; in TEST_F()
328 GrapplerItem item; in TEST_F()
358 GrapplerItem item; in TEST_F()
398 GrapplerItem item; in TEST_F()
426 GrapplerItem item; in TEST_F()
444 GrapplerItem item; in TEST_F()
462 GrapplerItem item; in TEST_F()
[all …]
Danalytical_cost_estimator.h34 struct GrapplerItem;
59 Status Initialize(const GrapplerItem& item) override;
70 const GrapplerItem* item_;
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmeta_optimizer.h39 Status Optimize(Cluster* cluster, const GrapplerItem& item,
42 void Feedback(Cluster* cluster, const GrapplerItem& item,
52 GrapplerItem* item) const;
Dmap_parallelization_test.cc31 Status OptimizeWithMapParallelization(const GrapplerItem& item, in OptimizeWithMapParallelization()
54 GrapplerItem item; in TEST_P()
85 GrapplerItem item; in TEST_P()
113 GrapplerItem item; in TEST()
Dbatch_parallelization_test.cc31 Status OptimizeWithBatchParallelization(const GrapplerItem& item, in OptimizeWithBatchParallelization()
52 GrapplerItem item; in TEST_P()
84 GrapplerItem item; in TEST_P()
114 GrapplerItem item; in TEST()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dtrt_optimization_pass.h57 const grappler::GrapplerItem& item,
60 void Feedback(grappler::Cluster* cluster, const grappler::GrapplerItem& item,
64 const grappler::GrapplerItem& item);
/external/tensorflow/tensorflow/core/grappler/clusters/
Dsingle_machine_test.cc76 GrapplerItem item; in TEST_F()
114 GrapplerItem item; in TEST_F()
127 GrapplerItem item; in TEST_F()
190 GrapplerItem item; in TEST_F()
241 GrapplerItem item; in TEST_F()
255 GrapplerItem item; in RunInfiniteTFLoop()
373 GrapplerItem item; in TEST_F()
411 GrapplerItem item; in TEST_F()
504 GrapplerItem CreateGrapplerItemWithResourceMemory() { in CreateGrapplerItemWithResourceMemory()
538 GrapplerItem item; in CreateGrapplerItemWithResourceMemory()
[all …]

12345678