Home
last modified time | relevance | path

Searched refs:DumpGraphToFile (Results 1 – 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/core/util/
Ddump_graph_test.cc35 string ret = DumpGraphToFile("graph", graph); in TEST()
37 ret = DumpGraphToFile("graph", graph); in TEST()
52 string ret = DumpGraphToFile("graph", graph); in TEST()
Ddump_graph.h51 string DumpGraphToFile(const string& name, Graph const& graph,
Ddump_graph.cc234 string DumpGraphToFile(const string& name, Graph const& graph, in DumpGraphToFile() function
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_control_flow.cc244 DumpGraphToFile( in FunctionalizeControlFlowForFunction()
250 DumpGraphToFile( in FunctionalizeControlFlowForFunction()
278 << DumpGraphToFile("functionalize_initial", *graph, library); in FunctionalizeControlFlow()
306 << DumpGraphToFile("functionalize_final", *graph, library); in FunctionalizeControlFlow()
330 DumpGraphToFile("functionalize_control_flow_before", *graph, in Run()
381 DumpGraphToFile("functionalize_control_flow_before_prune", *graph, in Run()
389 DumpGraphToFile("functionalize_control_flow_after", *graph, in Run()
Dfunctionalize_while.cc222 << DumpGraphToFile("functionalize_before", *graph, library); in FunctionalizeLoop()
414 << DumpGraphToFile("loop_condition", *cond_graph, library) in FunctionalizeLoop()
415 << " body: " << DumpGraphToFile("loop_body", *body_graph); in FunctionalizeLoop()
504 << DumpGraphToFile("functionalize_after", *graph, library); in FunctionalizeLoop()
Dgraph_compiler_util.cc160 VLOG(2) << "Post rewrite: " << DumpGraphToFile("tf2xla_post_rewrite", *graph); in RewriteAndPruneGraph()
163 VLOG(2) << "Post prune: " << DumpGraphToFile("tfcompile_post_prune", *graph); in RewriteAndPruneGraph()
Dxla_compiler.cc807 << DumpGraphToFile( in CompileFunction()
1312 << DumpGraphToFile(absl::StrCat("xla_compile_graph_", name), *graph, in CompileGraph()
/external/tensorflow/tensorflow/core/common_runtime/
Disolate_placer_inspection_required_ops_pass.cc38 DumpGraphToFile("isolate_deep_ops_before", *graph, nullptr, "/tmp"); in Run()
44 DumpGraphToFile("isolate_deep_ops_after", *graph, nullptr, "/tmp"); in Run()
Doptimization_registry.cc39 DumpGraphToFile( in RunGrouping()
46 DumpGraphToFile( in RunGrouping()
Dplacer.cc164 DumpGraphToFile("placer_input", *graph_, nullptr); in Run()
291 DumpGraphToFile("placer_output", *graph_, nullptr); in Run()
Dcontrol_flow_deps_to_chains.cc46 DumpGraphToFile("control_flow_deps_to_chains_before", *g, flib_def); in Run()
308 DumpGraphToFile("control_flow_deps_to_chains_after", *g, flib_def); in Run()
Dmemory_types.cc211 << DumpGraphToFile("EnsureMemoryTypes", *g); in EnsureMemoryTypes()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_configuration_rewrite_pass.cc249 DumpGraphToFile("distributed_tpu_configuration_before", *graph, in Run()
346 DumpGraphToFile("distributed_tpu_configuration_after", *graph, in Run()
361 DumpGraphToFile("distributed_tpu_shutdown_before", *graph, in Run()
395 DumpGraphToFile("distributed_tpu_shutdown_after", *graph, options.flib_def); in Run()
Dvariable_merger_pass.cc139 VLOG(1) << DumpGraphToFile("variable_merger_pass_before", *graph); in Run()
200 VLOG(1) << DumpGraphToFile("variable_merger_pass_after", *graph); in Run()
Dencapsulate_tpu_computations_pass.cc749 << DumpGraphToFile(absl::StrCat("move_head_oc_host_", xla_func_name), in MoveHeadOutsideCompilationToHost()
752 << DumpGraphToFile(absl::StrCat("move_head_oc_xla_", xla_func_name), in MoveHeadOutsideCompilationToHost()
983 << DumpGraphToFile( in RemoveUnusedXlaInput()
986 << DumpGraphToFile( in RemoveUnusedXlaInput()
1231 << DumpGraphToFile(absl::StrCat("move_tail_oc_host_", xla_func_name), in MoveTailOutsideCompilationToHost()
1234 << DumpGraphToFile(absl::StrCat("move_tail_oc_xla_", xla_func_name), in MoveTailOutsideCompilationToHost()
1356 << DumpGraphToFile( in ReplaceArgUsedByOutsideCompilationWithPlaceholder()
1359 << DumpGraphToFile( in ReplaceArgUsedByOutsideCompilationWithPlaceholder()
1466 << DumpGraphToFile( in RemoveUnusedXlaOutput()
1469 << DumpGraphToFile( in RemoveUnusedXlaOutput()
[all …]
Ddistributed_tpu_rewrite_pass.cc4827 VLOG(1) << DumpGraphToFile("distributed_tpu_graph_to_replicate", *computation, in RewriteTPUReplicateNode()
4992 VLOG(1) << DumpGraphToFile("distributed_tpu_compilation_before", *graph, in Run()
5025 VLOG(1) << DumpGraphToFile("distributed_tpu_compilation_after", *graph, in Run()
5090 VLOG(1) << DumpGraphToFile("distributed_tpu_compilation_after", *graph, in Run()
5096 VLOG(1) << DumpGraphToFile("host_loop_optimization_before", *graph, in Run()
5100 VLOG(1) << DumpGraphToFile("host_loop_optimization_after", *graph, in Run()
/external/tensorflow/tensorflow/core/tfrt/utils/
Dtfrt_graph_execution_state.cc161 DumpGraphToFile("after_pruning", *result.graph); in CreateOptimizedGraph()
176 DumpGraphToFile("after_functionalization", *result.graph); in CreateOptimizedGraph()
186 DumpGraphToFile("after_grappler", *result.graph); in CreateOptimizedGraph()
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_xla_computations_pass.cc370 << DumpGraphToFile("encapsulate_xla_computations_before", in Run()
381 << DumpGraphToFile("encapsulate_xla_computations_halfway", in Run()
387 << DumpGraphToFile("encapsulate_xla_computations_after", in Run()
Dincrease_dynamism_for_auto_jit_pass.cc379 DumpGraphToFile("before_increase_dynamism_for_auto_jit_pass", in Run()
386 DumpGraphToFile("increase_dynamism_for_auto_jit_pass", **options.graph, in Run()
Dencapsulate_subgraphs_pass.cc627 DumpGraphToFile(absl::StrCat("encapsulate_fdef_graph_", name), *graph_, in BuildFunctionDef()
650 DumpGraphToFile(absl::StrCat("replace_encapsulate_fdef_graph_", name), in ReplaceFunctionDef()
798 DumpGraphToFile( in SplitIntoSubgraphs()
1164 DumpGraphToFile("encapsulate_subgraphs_before", **options.graph, in Run()
1318 DumpGraphToFile("encapsulate_subgraphs_after", *graph_out, in Run()
Dmark_for_compilation_pass.cc854 DumpGraphToFile("before_mark_for_compilation", *graph_, flib_def_); in CreateClusters()
1445 DumpGraphToFile("mark_for_compilation", *graph_, flib_def_); in DumpPostClusteringGraphs()
1467 DumpGraphToFile("mark_for_compilation_annotated", new_graph, flib_def_); in DumpPostClusteringGraphs()
Dextract_outside_compilation_pass.cc1108 DumpGraphToFile(absl::StrCat("extract_outside_compilation_host_graph_for_", in ConstructHostGraph()
1309 DumpGraphToFile(shape_inference_graph_name, *g, fld); in RewriteShapeInferenceGraph()
2339 DumpGraphToFile( in ExtractOutsideCompilationForFunction()
2488 DumpGraphToFile( in ExtractOutsideCompilationForFunction()
2505 DumpGraphToFile("extract_outside_compilation_before", *g, fld); in ExtractOutsideCompilation()
2543 DumpGraphToFile("extract_outside_compilation_after", *g, fld); in ExtractOutsideCompilation()
Dbuild_xla_ops_pass.cc614 DumpGraphToFile("build_xla_ops", *graph, options.flib_def); in Run()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Ddump_graph_test.cc70 string ret = DumpGraphToFile("tir", graph); in TEST()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_functional_ops.cc1229 VLOG(1) << DumpGraphToFile("before_input_output_optimizations", *graph, in ComputeAsync()
1239 VLOG(1) << DumpGraphToFile( in ComputeAsync()
1249 VLOG(1) << DumpGraphToFile( in ComputeAsync()
1282 VLOG(1) << DumpGraphToFile("before_partition_helper.pbtxt", *graph, in ComputeAsync()
2100 VLOG(2) << DumpGraphToFile("before_enable_auto_xla_input_sharding", *graph, in OptimizeTpuInputOutputTensors()
2126 VLOG(2) << DumpGraphToFile("before_remove_descendant_nodes", *graph, in OptimizeTpuInputOutputTensors()
2151 VLOG(2) << DumpGraphToFile("before_get_input_output_info", *graph, in OptimizeTpuInputOutputTensors()
2158 VLOG(2) << DumpGraphToFile("before_optimize_tpu_input_output_tensors", *graph, in OptimizeTpuInputOutputTensors()
2197 VLOG(1) << DumpGraphToFile("optim_result", *graph); in OptimizeTpuInputOutputTensors()

12