| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | BUILD | 23 "//tensorflow/core/grappler:grappler_item",
24 "//tensorflow/core/grappler:op_types",
25 "//tensorflow/core/grappler:utils",
26 "//tensorflow/core/grappler/clusters:cluster",
27 "//tensorflow/core/grappler/costs:cost_estimator",
28 "//tensorflow/core/grappler/costs:graph_properties",
29 "//tensorflow/core/grappler/costs:op_level_cost_estimator",
30 "//tensorflow/core/grappler/costs:virtual_placer",
43 "//tensorflow/core/grappler:grappler_item",
44 "//tensorflow/core/grappler:utils",
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | BUILD | 7 "//tensorflow/core/grappler/optimizers/data:__subpackages__",
41 "//tensorflow/core/grappler:grappler_item",
42 "//tensorflow/core/grappler:mutable_graph_view",
43 "//tensorflow/core/grappler:op_types",
44 "//tensorflow/core/grappler/clusters:cluster",
45 "//tensorflow/core/grappler/optimizers:custom_graph_optimizer_registry",
48 "//tensorflow/core/grappler/utils:functions",
64 "//tensorflow/core/grappler:mutable_graph_view",
66 "//tensorflow/core/grappler:grappler_item",
67 "//tensorflow/core/grappler:op_types",
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | BUILD | 44 "//tensorflow/core/grappler/utils:functions",
45 "//tensorflow/core/grappler/utils:topological_sort",
46 "//tensorflow/core/grappler:mutable_graph_view",
47 "//tensorflow/core/grappler:op_types",
52 "//tensorflow/core/grappler:grappler_item",
53 "//tensorflow/core/grappler:utils",
54 "//tensorflow/core/grappler/clusters:cluster",
55 "//tensorflow/core/grappler/optimizers:evaluation_utils",
75 "//tensorflow/core/grappler:grappler_item",
76 "//tensorflow/core/grappler/clusters:single_machine",
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | BUILD | 18 "//tensorflow/core/grappler:op_types",
19 "//tensorflow/core/grappler:utils",
20 "//tensorflow/core/grappler/inputs:utils",
28 "//tensorflow/core/grappler/costs:graph_properties_testdata",
36 "//tensorflow/core/grappler:grappler_item",
37 "//tensorflow/core/grappler/clusters:virtual_cluster",
38 "//tensorflow/core/grappler/inputs:trivial_test_graph_input_yielder",
51 "//tensorflow/core/grappler:graph_topology_view",
52 "//tensorflow/core/grappler:graph_view",
53 "//tensorflow/core/grappler:op_types",
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | trt_optimization_pass.h | 33 class TRTOptimizationPass : public grappler::CustomGraphOptimizer {
53 Status Optimize(grappler::Cluster* cluster,
54 const grappler::GrapplerItem& item,
57 void Feedback(grappler::Cluster* cluster, const grappler::GrapplerItem& item,
60 void PrintDebugInfo(grappler::Cluster* cluster,
61 const grappler::GrapplerItem& item);
|
| D | trt_optimization_pass.cc | 81 void TRTOptimizationPass::PrintDebugInfo(grappler::Cluster* cluster, in PrintDebugInfo()
82 const grappler::GrapplerItem& item) { in PrintDebugInfo()
182 Status TRTOptimizationPass::Optimize(grappler::Cluster* cluster, in Optimize()
183 const grappler::GrapplerItem& item, in Optimize()
228 grappler::GraphProperties static_graph_properties(item); in Optimize()
274 void TRTOptimizationPass::Feedback(grappler::Cluster* cluster, in Feedback()
275 const grappler::GrapplerItem& item, in Feedback()
280 : public grappler::CustomGraphOptimizerRegistrar {
283 const grappler::CustomGraphOptimizerRegistry::Creator& cr, in VerboseCustomGraphOptimizerRegistrar()
285 : grappler::CustomGraphOptimizerRegistrar(cr, name) { in VerboseCustomGraphOptimizerRegistrar()
|
| D | convert_graph.h | 38 TrtCandidateSelector(const grappler::GraphProperties& graph_properties,
52 const grappler::GraphProperties& graph_properties_;
66 const grappler::GraphProperties* graph_properties = nullptr;
67 const grappler::Cluster* cluster = nullptr;
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | BUILD | 67 "//tensorflow/core/grappler:grappler_item",
84 "//tensorflow/core/grappler/costs:analytical_cost_estimator",
85 "//tensorflow/core/grappler/costs:op_level_cost_estimator",
86 "//tensorflow/core/grappler/costs:virtual_scheduler",
100 "//tensorflow/core/grappler:grappler_item",
101 "//tensorflow/core/grappler/inputs:trivial_test_graph_input_yielder",
123 "//tensorflow/core/grappler:utils",
146 "//tensorflow/core/grappler:grappler_item",
147 "//tensorflow/core/grappler:utils",
148 "//tensorflow/core/grappler/inputs:trivial_test_graph_input_yielder",
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | cluster.i | 23 std::shared_ptr<tensorflow::grappler::Cluster> cluster_;
117 GCluster(tensorflow::grappler::Cluster* cluster) : cluster_(cluster) {} in GCluster()
119 tensorflow::grappler::Cluster* operator->() const {
122 tensorflow::grappler::Cluster* get() const { in get()
129 std::shared_ptr<tensorflow::grappler::Cluster> cluster_;
135 int num_cpu_cores = tensorflow::grappler::GetNumAvailableLogicalCPUCores(); in TF_NewCluster()
136 int num_gpus = tensorflow::grappler::GetNumAvailableGPUs(); in TF_NewCluster()
138 tensorflow::grappler::Cluster* cluster_ = in TF_NewCluster()
139 new tensorflow::grappler::SingleMachine( in TF_NewCluster()
156 tensorflow::grappler::Cluster* cluster_ = in TF_NewVirtualCluster()
[all …]
|
| D | item.i | 37 std::shared_ptr<tensorflow::grappler::GrapplerItem> item_;
59 GItem(tensorflow::grappler::GrapplerItem* item) : item_(item) {} in GItem()
61 tensorflow::grappler::GrapplerItem* operator->() const {
64 const tensorflow::grappler::GrapplerItem& operator*() const {
70 std::shared_ptr<tensorflow::grappler::GrapplerItem> item_;
83 tensorflow::grappler::ItemConfig cfg; in TF_NewItem()
86 std::unique_ptr<tensorflow::grappler::GrapplerItem> item = in TF_NewItem()
87 tensorflow::grappler::GrapplerItemFromMetaGraphDef("item", meta_graph, cfg); in TF_NewItem()
122 tensorflow::Status s = tensorflow::grappler::TopologicalSort(&subgraph); in TF_IdentifyImportantOps()
147 tensorflow::grappler::GraphProperties properties(*item); in TF_GetOpProperties()
[all …]
|
| D | model_analyzer.i | 45 tensorflow::grappler::ItemConfig cfg; in GenerateModelReport()
47 std::unique_ptr<tensorflow::grappler::GrapplerItem> item = in GenerateModelReport()
48 tensorflow::grappler::GrapplerItemFromMetaGraphDef("metagraph", metagraph, cfg); in GenerateModelReport()
54 tensorflow::grappler::ModelAnalyzer analyzer(*item); in GenerateModelReport()
|
| D | cost_analyzer.i | 48 tensorflow::grappler::ItemConfig cfg; in GenerateCostReport()
50 std::unique_ptr<tensorflow::grappler::GrapplerItem> item = in GenerateCostReport()
51 tensorflow::grappler::GrapplerItemFromMetaGraphDef("metagraph", metagraph, cfg); in GenerateCostReport()
57 tensorflow::grappler::CostAnalyzer analyzer(*item, cluster.get(), suffix); in GenerateCostReport()
|
| D | tf_optimizer.i | 85 prop = tensorflow::grappler::GetDeviceInfo(device->parsed_name()); in DetectDevices()
99 tensorflow::grappler::ItemConfig item_config; in TF_OptimizeGraph()
102 std::unique_ptr<tensorflow::grappler::GrapplerItem> grappler_item = in TF_OptimizeGraph()
103 tensorflow::grappler::GrapplerItemFromMetaGraphDef(graph_id, metagraph, item_config); in TF_OptimizeGraph()
112 tensorflow::grappler::MetaOptimizer optimizer(cpu_device, config_proto); in TF_OptimizeGraph()
|
| D | graph_placer.py | 26 from tensorflow.python.grappler import cluster as gcluster
27 from tensorflow.python.grappler import hierarchical_controller
28 from tensorflow.python.grappler import item as gitem
29 from tensorflow.python.grappler import tf_optimizer
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_fusion_optimizer.h | 25 class XlaFusionOptimizer : public grappler::CustomGraphOptimizer {
37 Status Optimize(grappler::Cluster* cluster,
38 const grappler::GrapplerItem& item,
41 void Feedback(grappler::Cluster* cluster, const grappler::GrapplerItem& item, in Feedback()
|
| D | xla_fusion_optimizer_test.cc | 32 class XlaFusionOptimizerTest : public grappler::GrapplerTest {
61 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()
|
| /external/tensorflow/tensorflow/contrib/cmake/ |
| D | tf_grappler.cmake | 19 "${tensorflow_source_dir}/tensorflow/core/grappler/clusters/single_machine.cc"
20 "${tensorflow_source_dir}/tensorflow/core/grappler/clusters/single_machine.h"
21 "${tensorflow_source_dir}/tensorflow/python/grappler/cost_analyzer.cc"
22 "${tensorflow_source_dir}/tensorflow/python/grappler/cost_analyzer.h"
23 "${tensorflow_source_dir}/tensorflow/python/grappler/model_analyzer.cc"
24 "${tensorflow_source_dir}/tensorflow/python/grappler/model_analyzer.h"
|
| D | tf_core_cpu.cmake | 29 "${tensorflow_source_dir}/tensorflow/core/grappler/*.h"
30 "${tensorflow_source_dir}/tensorflow/core/grappler/*.cc"
31 "${tensorflow_source_dir}/tensorflow/core/grappler/*/*.h"
32 "${tensorflow_source_dir}/tensorflow/core/grappler/*/*.cc"
63 "${tensorflow_source_dir}/tensorflow/core/grappler/clusters/single_machine.h"
64 "${tensorflow_source_dir}/tensorflow/core/grappler/clusters/single_machine.cc"
65 "${tensorflow_source_dir}/tensorflow/core/grappler/inputs/trivial_test_graph_input_yielder.h"
66 "${tensorflow_source_dir}/tensorflow/core/grappler/inputs/trivial_test_graph_input_yielder.cc"
82 "${tensorflow_source_dir}/tensorflow/core/grappler/devices.h"
83 "${tensorflow_source_dir}/tensorflow/core/grappler/devices.cc"
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/ |
| D | BUILD | 70 "//tensorflow/core/grappler/costs:graph_properties",
258 "//tensorflow/core/grappler/clusters:cluster",
259 "//tensorflow/core/grappler/optimizers:custom_graph_optimizer",
260 "//tensorflow/core/grappler/optimizers:custom_graph_optimizer_registry",
261 "//tensorflow/core/grappler:grappler_item",
262 "//tensorflow/core/grappler:utils",
269 "//tensorflow/core/grappler:devices",
270 "//tensorflow/core/grappler/clusters:virtual_cluster",
271 "//tensorflow/core/grappler/costs:graph_properties",
272 "//tensorflow/core/grappler/optimizers:meta_optimizer",
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/graph_analyzer/ |
| D | BUILD | 27 "//tensorflow/core/grappler:op_types",
28 "//tensorflow/core/grappler:utils",
45 "//tensorflow/core/grappler:grappler_item",
64 "//tensorflow/core/grappler:op_types",
105 "//tensorflow/core/grappler:utils",
|
| /external/tensorflow/tensorflow/core/grappler/inputs/ |
| D | BUILD | 58 "//tensorflow/core/grappler:grappler_item",
74 "//tensorflow/core/grappler:grappler_item",
75 "//tensorflow/core/grappler:grappler_item_builder",
76 "//tensorflow/core/grappler:utils",
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | graph_rewrite_dataset.cc | 105 tensorflow::grappler::function_utils::SetUniqueFunctionNodeName( in AddFakeSinks()
142 tensorflow::grappler::graph_utils::SetUniqueGraphNodeName("Sink", graph_def, in ApplyOptimizations()
168 tensorflow::grappler::ItemConfig item_config; in ApplyOptimizations()
170 std::unique_ptr<tensorflow::grappler::GrapplerItem> grappler_item = in ApplyOptimizations()
171 tensorflow::grappler::GrapplerItemFromMetaGraphDef( in ApplyOptimizations()
176 tensorflow::grappler::VirtualCluster cluster(device_map); in ApplyOptimizations()
182 TF_RETURN_IF_ERROR(tensorflow::grappler::RunMetaOptimizer( in ApplyOptimizations()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/ |
| D | vectorizer_registry.h | 26 namespace grappler {
63 static ::tensorflow::grappler::vectorizer_registration:: \
65 op_type, ::std::unique_ptr<::tensorflow::grappler::Vectorizer>( \
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | BUILD | 124 "//tensorflow/core/grappler/inputs:trivial_test_graph_input_yielder",
165 "//tensorflow/core/grappler/inputs:utils",
166 "//tensorflow/core/grappler/optimizers:model_pruner",
179 "//tensorflow/core/grappler/inputs:trivial_test_graph_input_yielder",
198 "//tensorflow/core/grappler/inputs:trivial_test_graph_input_yielder",
238 "//tensorflow/core/grappler/inputs:trivial_test_graph_input_yielder",
|
| /external/tensorflow/tensorflow/core/grappler/verifiers/ |
| D | BUILD | 31 "//tensorflow/core/grappler/utils:topological_sort",
46 "//tensorflow/core/grappler:grappler_item",
47 "//tensorflow/core/grappler/inputs:trivial_test_graph_input_yielder",
|