| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | BUILD | 42 "//tensorflow/core/grappler:grappler_item",
43 "//tensorflow/core/grappler:op_types",
44 "//tensorflow/core/grappler:utils",
45 "//tensorflow/core/grappler/clusters:cluster",
46 "//tensorflow/core/grappler/costs:cost_estimator",
47 "//tensorflow/core/grappler/costs:graph_properties",
48 "//tensorflow/core/grappler/costs:op_level_cost_estimator",
49 "//tensorflow/core/grappler/costs:virtual_placer",
62 "//tensorflow/core/grappler:grappler_item",
63 "//tensorflow/core/grappler:utils",
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | BUILD | 7 "//tensorflow/core/grappler/optimizers/data:__subpackages__",
52 "//tensorflow/core/grappler:grappler_item",
53 "//tensorflow/core/grappler:mutable_graph_view",
54 "//tensorflow/core/grappler:op_types",
55 "//tensorflow/core/grappler/clusters:cluster",
56 "//tensorflow/core/grappler/optimizers:custom_graph_optimizer_registry",
59 "//tensorflow/core/grappler/utils:functions",
72 "//tensorflow/core/grappler:mutable_graph_view",
76 "//tensorflow/core/grappler:grappler_item",
77 "//tensorflow/core/grappler:op_types",
[all …]
|
| /external/tensorflow/tensorflow/tools/def_file_filter/ |
| D | symbols_pybind.txt | 49 [//tensorflow/core/grappler/graph_analyzer:graph_analyzer_tool] # graph_analyze
50 tensorflow::grappler::graph_analyzer::GraphAnalyzerTool
264 [//tensorflow/core/grappler:grappler_item] # tf_item
265 tensorflow::grappler::GrapplerItem::MainOpsFanin
266 tensorflow::grappler::GrapplerItem::EnqueueOpsFanin
268 [//tensorflow/core/grappler/costs:graph_properties] # tf_item
269 tensorflow::grappler::GraphProperties::InferStatically
270 tensorflow::grappler::GraphProperties::GetOutputProperties
272 [//tensorflow/core/grappler:grappler_item_builder] # tf_item
273 tensorflow::grappler::GrapplerItemFromMetaGraphDef
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | BUILD | 21 actual = "//tensorflow/core/grappler/costs/graph_properties_testdata:graph_properties_testdata",
76 "//tensorflow/core/grappler/utils:functions",
77 "//tensorflow/core/grappler/utils:topological_sort",
78 "//tensorflow/core/grappler:mutable_graph_view",
79 "//tensorflow/core/grappler:op_types",
84 "//tensorflow/core/grappler:grappler_item",
85 "//tensorflow/core/grappler:utils",
86 "//tensorflow/core/grappler/clusters:cluster",
87 "//tensorflow/core/grappler/optimizers:evaluation_utils",
112 "//tensorflow/core/grappler:grappler_item",
[all …]
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | cluster_wrapper.cc | 53 const tensorflow::grappler::GrapplerItem& item, in _GetOpPerformanceDataAndRunTime()
54 tensorflow::grappler::CostEstimator* cost_measure, in _GetOpPerformanceDataAndRunTime()
56 tensorflow::grappler::Costs* costs) { in _GetOpPerformanceDataAndRunTime()
65 *op_performance_data = tensorflow::grappler::CostGraphToOpPerformanceData( in _GetOpPerformanceDataAndRunTime()
71 PYBIND11_MAKE_OPAQUE(tensorflow::grappler::Cluster);
74 py::class_<tensorflow::grappler::Cluster> grappler_cluster( in PYBIND11_MODULE()
79 bool disable_detailed_stats) -> tensorflow::grappler::Cluster* { in PYBIND11_MODULE()
83 tensorflow::grappler::GetNumAvailableLogicalCPUCores(); in PYBIND11_MODULE()
84 int num_gpus = tensorflow::grappler::GetNumAvailableGPUs(); in PYBIND11_MODULE()
86 std::unique_ptr<tensorflow::grappler::Cluster> cluster = in PYBIND11_MODULE()
[all …]
|
| D | item_wrapper.cc | 122 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()
142 tensorflow::grappler::ItemConfig cfg; in PYBIND11_MODULE()
145 std::unique_ptr<tensorflow::grappler::GrapplerItem> item = in PYBIND11_MODULE()
146 tensorflow::grappler::GrapplerItemFromMetaGraphDef( in PYBIND11_MODULE()
156 [](tensorflow::grappler::GrapplerItem* item, in PYBIND11_MODULE()
179 tensorflow::grappler::TopologicalSort(&subgraph)); in PYBIND11_MODULE()
192 [](tensorflow::grappler::GrapplerItem* item) in PYBIND11_MODULE()
194 tensorflow::grappler::GraphProperties properties(*item); in PYBIND11_MODULE()
[all …]
|
| D | cost_analyzer_wrapper.cc | 33 bool verbose, tensorflow::grappler::Cluster* cluster) -> py::bytes { in PYBIND11_MODULE()
40 tensorflow::grappler::ItemConfig cfg; in PYBIND11_MODULE()
42 std::unique_ptr<tensorflow::grappler::GrapplerItem> item = in PYBIND11_MODULE()
43 tensorflow::grappler::GrapplerItemFromMetaGraphDef( in PYBIND11_MODULE()
51 tensorflow::grappler::CostAnalyzer analyzer(*item, cluster, suffix); in PYBIND11_MODULE()
|
| D | tf_optimizer_wrapper.cc | 51 prop = tensorflow::grappler::GetDeviceInfo(device->parsed_name()); in DetectDevices()
63 [](tensorflow::grappler::Cluster* cluster, in PYBIND11_MODULE()
81 tensorflow::grappler::ItemConfig item_config; in PYBIND11_MODULE()
86 std::unique_ptr<tensorflow::grappler::GrapplerItem> grappler_item = in PYBIND11_MODULE()
87 tensorflow::grappler::GrapplerItemFromMetaGraphDef( in PYBIND11_MODULE()
96 tensorflow::grappler::MetaOptimizer optimizer(cpu_device, config_proto); in PYBIND11_MODULE()
|
| D | model_analyzer_wrapper.cc | 37 tensorflow::grappler::ItemConfig cfg; in PYBIND11_MODULE()
39 std::unique_ptr<tensorflow::grappler::GrapplerItem> item = in PYBIND11_MODULE()
40 tensorflow::grappler::GrapplerItemFromMetaGraphDef( in PYBIND11_MODULE()
47 tensorflow::grappler::ModelAnalyzer analyzer(*item); in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | BUILD | 32 "//tensorflow/core/grappler:op_types",
33 "//tensorflow/core/grappler:utils",
34 "//tensorflow/core/grappler/inputs:utils",
42 "//tensorflow/core/grappler/costs:graph_properties_testdata",
50 "//tensorflow/core/grappler:grappler_item",
51 "//tensorflow/core/grappler/clusters:virtual_cluster",
52 "//tensorflow/core/grappler/inputs:trivial_test_graph_input_yielder",
65 "//tensorflow/core/grappler:graph_topology_view",
66 "//tensorflow/core/grappler:graph_view",
67 "//tensorflow/core/grappler:op_types",
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | trt_optimization_pass.h | 32 class TRTOptimizationPass : public grappler::CustomGraphOptimizer {
56 Status Optimize(grappler::Cluster* cluster,
57 const grappler::GrapplerItem& item,
60 void Feedback(grappler::Cluster* cluster, const grappler::GrapplerItem& item,
63 void PrintDebugInfo(grappler::Cluster* cluster,
64 const grappler::GrapplerItem& item);
|
| D | trt_optimization_pass.cc | 82 void TRTOptimizationPass::PrintDebugInfo(grappler::Cluster* cluster, in PrintDebugInfo()
83 const grappler::GrapplerItem& item) { in PrintDebugInfo()
185 Status TRTOptimizationPass::Optimize(grappler::Cluster* cluster, in Optimize()
186 const grappler::GrapplerItem& item, in Optimize()
251 void TRTOptimizationPass::Feedback(grappler::Cluster* cluster, in Feedback()
252 const grappler::GrapplerItem& item, in Feedback()
257 : public grappler::CustomGraphOptimizerRegistrar {
260 const grappler::CustomGraphOptimizerRegistry::Creator& cr, in VerboseCustomGraphOptimizerRegistrar()
262 : grappler::CustomGraphOptimizerRegistrar(cr, name) { in VerboseCustomGraphOptimizerRegistrar()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/inference/ |
| D | batch_op_rewriter.h | 24 namespace grappler {
33 class BatchOpRewriter : public ::tensorflow::grappler::CustomGraphOptimizer {
43 ::tensorflow::grappler::Cluster* cluster,
44 const ::tensorflow::grappler::GrapplerItem& item,
47 void Feedback(::tensorflow::grappler::Cluster* cluster, in Feedback()
48 const ::tensorflow::grappler::GrapplerItem& item, in Feedback()
|
| D | BUILD | 5 "//tensorflow/core/grappler/optimizers/inference:__subpackages__",
30 "//tensorflow/core/grappler:grappler_item",
31 "//tensorflow/core/grappler/optimizers:custom_graph_optimizer",
32 "//tensorflow/core/grappler/optimizers:custom_graph_optimizer_registry",
51 "//tensorflow/core/grappler:grappler_item",
52 "//tensorflow/core/grappler/optimizers:graph_optimizer",
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | BUILD | 81 "//tensorflow/core/grappler:grappler_item",
98 "//tensorflow/core/grappler/costs:analytical_cost_estimator",
99 "//tensorflow/core/grappler/costs:op_level_cost_estimator",
100 "//tensorflow/core/grappler/costs:virtual_scheduler",
114 "//tensorflow/core/grappler:grappler_item",
115 "//tensorflow/core/grappler/inputs:trivial_test_graph_input_yielder",
137 "//tensorflow/core/grappler:utils",
162 "//tensorflow/core/grappler:grappler_item",
163 "//tensorflow/core/grappler:utils",
164 "//tensorflow/core/grappler/inputs:trivial_test_graph_input_yielder",
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | compute_batch_size_op.cc | 30 using grappler::graph_utils::GetScalarConstNodeValue;
74 const grappler::GraphView& graph, in GetInputNode()
77 grappler::GraphView::InputPort input_port = in GetInputNode()
103 grappler::GraphView graph_view(&graph_def); in Compute()
116 const grappler::GraphView& graph) { in GetBatchSizeFromBatchNode()
153 int64 GetBatchSize(const NodeDef& node, const grappler::GraphView& graph) { in GetBatchSize()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | rewrite_utils.cc | 54 tensorflow::grappler::function_utils::SetUniqueFunctionNodeName( in AddFakeSinks()
90 tensorflow::grappler::graph_utils::SetUniqueGraphNodeName("Sink", graph_def, in ApplyRewrites()
117 tensorflow::grappler::ItemConfig item_config; in ApplyRewrites()
119 std::unique_ptr<tensorflow::grappler::GrapplerItem> grappler_item = in ApplyRewrites()
120 tensorflow::grappler::GrapplerItemFromMetaGraphDef( in ApplyRewrites()
126 tensorflow::grappler::VirtualCluster cluster(device_map); in ApplyRewrites()
131 TF_RETURN_IF_ERROR(tensorflow::grappler::RunMetaOptimizer( in ApplyRewrites()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/ |
| D | BUILD | 119 "//tensorflow/core/grappler/costs:graph_properties",
304 "//tensorflow/core/grappler:op_types",
406 "//tensorflow/core/grappler/clusters:cluster",
407 "//tensorflow/core/grappler/optimizers:custom_graph_optimizer",
408 "//tensorflow/core/grappler/optimizers:custom_graph_optimizer_registry",
409 "//tensorflow/core/grappler:grappler_item",
410 "//tensorflow/core/grappler:op_types",
411 "//tensorflow/core/grappler:utils",
419 "//tensorflow/core/grappler:devices",
420 "//tensorflow/core/grappler/clusters:virtual_cluster",
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/graph_analyzer/ |
| D | BUILD | 30 "//tensorflow/core/grappler:op_types",
31 "//tensorflow/core/grappler:utils",
48 "//tensorflow/core/grappler/utils:transitive_fanin",
66 "//tensorflow/core/grappler:op_types",
107 "//tensorflow/core/grappler:utils",
|
| /external/tensorflow/tensorflow/core/grappler/inputs/ |
| D | BUILD | 61 "//tensorflow/core/grappler:grappler_item",
77 "//tensorflow/core/grappler:grappler_item",
78 "//tensorflow/core/grappler:grappler_item_builder",
79 "//tensorflow/core/grappler:utils",
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | upgrade_graph.cc | 152 grappler::ItemConfig item_config; in RunGrappler()
154 std::unique_ptr<grappler::GrapplerItem> item = in RunGrappler()
155 grappler::GrapplerItemFromMetaGraphDef("graph", meta_graph_def, in RunGrappler()
170 grappler::VirtualCluster cluster(&dev_set); in RunGrappler()
174 TF_RETURN_IF_ERROR(grappler::RunMetaOptimizer( in RunGrappler()
|
| /external/tensorflow/tensorflow/core/profiler/utils/ |
| D | cost_utils.cc | 80 grappler::DeviceInfo TfOpRoofLineCostEstimator::GetDeviceInfo( in GetDeviceInfo()
84 return grappler::DeviceInfo(/*gigaops=*/1, /*gb_per_sec=*/1); in GetDeviceInfo()
109 grappler::OpContext op_context; in Predict()
115 grappler::Costs costs = PredictCosts(op_context); in Predict()
|
| /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 | 144 "//tensorflow/core/grappler/inputs:trivial_test_graph_input_yielder",
162 "//tensorflow/core/grappler/utils:transitive_fanin",
186 "//tensorflow/core/grappler/inputs:utils",
187 "//tensorflow/core/grappler/optimizers:model_pruner",
200 "//tensorflow/core/grappler/inputs:trivial_test_graph_input_yielder",
219 "//tensorflow/core/grappler/inputs:trivial_test_graph_input_yielder",
259 "//tensorflow/core/grappler/inputs:trivial_test_graph_input_yielder",
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
| D | segment_test.cc | 63 const grappler::GraphProperties* graph_properties, in RunTest()
387 grappler::GrapplerItem item; in TEST_F()
391 grappler::GraphProperties static_graph_properties(item); in TEST_F()
419 grappler::GrapplerItem item; in TEST_F()
423 grappler::GraphProperties static_graph_properties(item); in TEST_F()
447 grappler::GrapplerItem item; in TEST_F()
452 grappler::GraphProperties static_graph_properties(item); in TEST_F()
476 grappler::GrapplerItem item; in TEST_F()
481 grappler::GraphProperties static_graph_properties(item); in TEST_F()
514 grappler::GrapplerItem item; in TEST_F()
[all …]
|