Home
last modified time | relevance | path

Searched refs:OptimizeGraph (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_optimizer.h93 void OptimizeGraph(FunctionLibraryRuntime* lib, std::unique_ptr<Graph>* g,
95 void OptimizeGraph(FunctionLibraryRuntime* lib, std::unique_ptr<Graph>* g);
Dgraph_optimizer.cc160 void OptimizeGraph(FunctionLibraryRuntime* lib, std::unique_ptr<Graph>* g, in OptimizeGraph() function
171 void OptimizeGraph(FunctionLibraryRuntime* lib, std::unique_ptr<Graph>* g) { in OptimizeGraph() function
172 OptimizeGraph(lib, g, GraphOptimizer::Options()); in OptimizeGraph()
Dgraph_execution_state.h141 Status OptimizeGraph(
Dgraph_execution_state.cc645 Status GraphExecutionState::OptimizeGraph( in OptimizeGraph() function in tensorflow::GraphExecutionState
868 Status s = OptimizeGraph(options, *graph_, flib_def_.get(), &optimized_graph, in BuildGraph()
Dfunction_test.cc1405 TEST_F(FunctionLibraryRuntimeTest, OptimizeGraph) { in TEST_F() argument
1411 OptimizeGraph(flr0_, &g); in TEST_F()
1456 OptimizeGraph(flr0_, &g); in TEST_F()
1489 OptimizeGraph(flr0_, &g); in TEST_F()
1627 OptimizeGraph(flr0_, &g); in TEST_F()
1881 OptimizeGraph(flr0_, &g); in TEST_F()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmeta_optimizer.h94 Status OptimizeGraph(
97 Status OptimizeGraph(Cluster* cluster, GrapplerItem&& item,
154 Status OptimizeGraph(
Dmeta_optimizer.cc746 Status MetaOptimizer::OptimizeGraph( in OptimizeGraph() function in tensorflow::grappler::MetaOptimizer
892 Status MetaOptimizer::OptimizeGraph(Cluster* cluster, GrapplerItem&& item, in OptimizeGraph() function in tensorflow::grappler::MetaOptimizer
904 return OptimizeGraph(std::move(optimizers), cluster, std::move(item), in OptimizeGraph()
1077 OptimizeGraph(cluster, GrapplerItem(item), optimized_graph)); in OptimizeConsumeItem()
1244 TF_RETURN_IF_ERROR(OptimizeGraph(cluster, std::move(func_item_copy), in OptimizeConsumeItem()
1292 TF_RETURN_IF_ERROR(OptimizeGraph(optimizers, cluster, std::move(tfg_item), in OptimizeConsumeItem()
1367 Status OptimizeGraph( in OptimizeGraph() function
/external/tensorflow/tensorflow/python/grappler/
Dmemory_optimizer_test.py57 graph = tf_optimizer.OptimizeGraph(config, mg)
86 graph = tf_optimizer.OptimizeGraph(config, mg)
146 rewritten_graph_def = tf_optimizer.OptimizeGraph(config, original_metagraph)
176 rewritten_graph_def = tf_optimizer.OptimizeGraph(config, original_metagraph)
205 rewritten_graph_def = tf_optimizer.OptimizeGraph(config, original_metagraph)
316 rewritten_graph_def = tf_optimizer.OptimizeGraph(config, metagraph)
Dtf_optimizer_test.py53 graph = tf_optimizer.OptimizeGraph(config, mg)
79 optimized_graph = tf_optimizer.OptimizeGraph(config, mg)
124 optimized_graph = tf_optimizer.OptimizeGraph(config, mg)
Dtf_optimizer.py27 def OptimizeGraph(config_proto, function
Dcost_analyzer_tool.py83 optimized_graph = tf_optimizer.OptimizeGraph(config, metagraph)
/external/tensorflow/tensorflow/python/kernel_tests/signal/
Dtest_util.py46 return tf_optimizer.OptimizeGraph(config_proto, metagraph)
/external/tensorflow/tensorflow/core/tfrt/utils/
Dtfrt_graph_execution_state.h103 StatusOr<std::unique_ptr<tensorflow::Graph>> OptimizeGraph(
Dtfrt_graph_execution_state.cc454 OptimizeGraph(*result.graph, build_graph_options); in CreateOptimizedGraph()
812 auto status = graph_execution_state->OptimizeGraph( in OptimizeFunctions()
840 TfrtGraphExecutionState::OptimizeGraph( in OptimizeGraph() function in tensorflow::tfrt_stub::TfrtGraphExecutionState
849 TF_RETURN_IF_ERROR(graph_execution_state_->OptimizeGraph( in OptimizeGraph()
/external/tensorflow/tensorflow/core/tfrt/eager/
Dtransform_graph_function.cc158 auto status = grappler::OptimizeGraph( in TransformGraphFunction()
/external/tensorflow/tensorflow/core/tpu/
Dtpu_compile.cc164 Status OptimizeGraph(const tpu::TPUCompileMetadataProto& metadata, in OptimizeGraph() function
438 TF_RETURN_IF_ERROR(OptimizeGraph(metadata, partial_arg_shapes, &graph, in CompileTFFunctionToHlo()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dauto_shard.cc180 Status OptimizeGraph(const GrapplerItem& item, int64_t num_workers,
788 Status OptimizeGraph(const GrapplerItem& item, int64_t num_workers, in OptimizeGraph() function
946 TF_RETURN_IF_ERROR(OptimizeGraph(item, num_workers_, index_, in OptimizeAndCollectStats()
/external/tensorflow/tensorflow/core/kernels/
Dpartitioned_function_ops.cc196 std::bind(grappler::OptimizeGraph, std::placeholders::_1, in Instantiate()
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py554 self._converted_graph_def = tf_optimizer.OptimizeGraph(
969 new_graph_def = tf_optimizer.OptimizeGraph(config, meta_graph)
1281 return tf_optimizer.OptimizeGraph(
/external/tensorflow/tensorflow/python/tools/
Dsaved_model_aot_compile.py429 return tf_optimizer.OptimizeGraph(config, new_meta_graph_def)
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.cc204 grappler::OptimizeGraph, std::placeholders::_1, std::placeholders::_2, in InstantiateFunc()
/external/tensorflow/tensorflow/lite/python/
Dutil.py259 return tf_optimizer.OptimizeGraph(config, meta_graph)
/external/tensorflow/tensorflow/python/framework/
Dconvert_to_constants.py1052 return tf_optimizer.OptimizeGraph(config, meta_graph)
/external/pytorch/torch/csrc/jit/runtime/static/
Dimpl.cpp156 void OptimizeGraph( in OptimizeGraph() function
273 OptimizeGraph(graph, opts, std::move(sample_inputs)); in PrepareGraphForStaticModule()
/external/tensorflow/tensorflow/core/data/
Dcaptured_function.cc652 std::bind(tensorflow::grappler::OptimizeGraph, std::placeholders::_1, in Instantiate()

12