Home
last modified time | relevance | path

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

123456789

/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,
96 Cluster* cluster, GrapplerItem&& item, GraphDef* optimized_graph);
97 Status OptimizeGraph(Cluster* cluster, GrapplerItem&& item,
118 GrapplerItem* optimized_item, GraphDef* optimized_graph,
132 Status RunMetaOptimizer(GrapplerItem&& item, const ConfigProto& cfg,
159 const GrapplerItem::OptimizationOptions& optimization_options,
Dmeta_optimizer_test.cc59 Status Optimize(Cluster* cluster, const GrapplerItem& item, in Optimize()
96 gtl::FlatMap<string, GrapplerItem::OptimizationOptions>* in SetOptimizationOptions()
113 Status Optimize(Cluster* cluster, const GrapplerItem& item, in Optimize()
123 static gtl::FlatMap<string, GrapplerItem::OptimizationOptions>*
127 gtl::FlatMap<string, GrapplerItem::OptimizationOptions>*
136 GrapplerItem item; in TEST_F()
155 GrapplerItem item; in TEST_F()
176 GrapplerItem item; in TEST_F()
199 GrapplerItem item; in TEST_F()
223 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 …]
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 …]
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 …]
Dgraph_optimizer.h31 struct GrapplerItem;
58 virtual Status Optimize(Cluster* cluster, const GrapplerItem& item,
62 virtual Status Optimize(Cluster* cluster, GrapplerItem&& item, in Optimize()
Dauto_parallel.h38 Status Optimize(Cluster* cluster, const GrapplerItem& item,
47 const GrapplerItem* item_;
50 Status Initialize(const GrapplerItem& item);
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 …]
Dgeneric_layout_optimizer_test.cc235 GrapplerItem item; in TEST_F()
261 GrapplerItem item; in TEST_F()
281 GrapplerItem item; in TEST_F()
304 GrapplerItem item; in TEST_F()
323 GrapplerItem item; in TEST_F()
346 GrapplerItem item; in TEST_F()
368 GrapplerItem item; in TEST_F()
402 GrapplerItem item; in TEST_F()
428 GrapplerItem item; in TEST_F()
481 GrapplerItem item; in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dauto_shard_test.cc47 void FinishItem(GrapplerItem* item, const string& input_node_name) { in FinishItem()
75 GrapplerItem item; in TEST()
105 GrapplerItem item; in TEST()
137 GrapplerItem item; in TEST()
170 GrapplerItem item; in TEST()
196 GrapplerItem item; in TEST()
223 GrapplerItem item; in TEST()
250 GrapplerItem item; in TEST()
276 GrapplerItem item; in TEST()
300 GrapplerItem item; in TEST()
[all …]
Dinject_prefetch_test.cc37 Status Optimize(InjectPrefetch &optimizer, const GrapplerItem &item, in Optimize()
49 Status OptimizeWithInjectPrefetch(const GrapplerItem &item, GraphDef *output, in OptimizeWithInjectPrefetch()
61 GrapplerItem item; in TEST_P()
87 GrapplerItem item; in TEST()
105 GrapplerItem item; in TEST()
125 GrapplerItem item; in TEST()
148 GrapplerItem item; in TEST()
Dmake_sloppy_test.cc34 GrapplerItem item; in TEST()
62 GrapplerItem item; in TEST()
88 GrapplerItem item; in TEST()
112 GrapplerItem item; in TEST()
140 GrapplerItem item; in TEST()
166 GrapplerItem item; in TEST()
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()
Dfilter_parallelization_test.cc31 Status OptimizeWithFilterParallelization(const GrapplerItem& item, in OptimizeWithFilterParallelization()
54 GrapplerItem item; in TEST_P()
86 GrapplerItem item; in TEST_P()
114 GrapplerItem item; in TEST()
Dbatch_parallelization_test.cc31 Status OptimizeWithBatchParallelization(const GrapplerItem& item, in OptimizeWithBatchParallelization()
52 GrapplerItem item; in TEST_P()
85 GrapplerItem item; in TEST_P()
119 GrapplerItem item; in TEST_P()
Dmap_and_filter_fusion_test.cc37 GrapplerItem item; in TEST()
63 GrapplerItem item; in TEST()
99 GrapplerItem item; in TEST()
128 GrapplerItem item; in TEST()
/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()
239 GrapplerItem item; in TEST_F()
326 GrapplerItem item; in RunTestAndValidate()
447 GrapplerItem item; in TEST_F()
496 GrapplerItem item; in TEST_F()
526 GrapplerItem item; in TEST_F()
566 GrapplerItem item; in TEST_F()
594 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/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/clusters/
Dsingle_machine_test.cc75 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 …]

123456789