Home
last modified time | relevance | path

Searched defs:outside_compilation_attr_name (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_util.cc74 Graph* g, const string& outside_compilation_attr_name) { in PreprocessControlEdgesBetweenOutsideCompilations()
114 Graph* g, const string& outside_compilation_attr_name) { in PreprocessDataEdgesBetweenOutsideCompilations()
200 Graph* g, const string& outside_compilation_attr_name) { in PostprocessDataEdgesBetweenOutsideCompilations()
276 Graph* g, const string& outside_compilation_attr_name) { in PostprocessControlEdgesBetweenOutsideCompilations()
348 const Graph* g, const string& outside_compilation_attr_name) { in OutsideCompilationClusterDependencies()
384 Graph* g, const string& outside_compilation_attr_name) { in PreprocessEdgesBetweenOutsideCompilations()
410 Graph* g, const string& outside_compilation_attr_name) { in PostprocessEdgesBetweenOutsideCompilations()
Dextract_outside_compilation_pass.h48 const string& outside_compilation_attr_name, in RewriteOutsideCompilationSubgraphFn()
Dextract_outside_compilation_pass.cc982 const string& xla_cluster_name, const string& outside_compilation_attr_name, in ConstructHostGraph()
1393 const string& outside_compilation_attr_name, const string& xla_cluster_name, in BuildHostGraphForIfNode()
1553 const string& xla_cluster_name, const string& outside_compilation_attr_name, in RewriteHostWhileLoopCond()
1629 const string& xla_cluster_name, const string& outside_compilation_attr_name, in RewriteHostWhileLoopBody()
1688 const string& outside_compilation_attr_name, const string& xla_cluster_name, in BuildHostGraphForWhileNode()
1757 const string& outside_compilation_attr_name, in BuildHostGraphForFuncCallNode()
1807 const string& outside_compilation_attr_name, const string& xla_cluster_name, in ExtractOutsideCompilationForFuncCallNode()
1893 const string& outside_compilation_attr_name, const string& xla_cluster_name, in ExtractOutsideCompilationForIfNode()
2012 const string& outside_compilation_attr_name, const string& xla_cluster_name, in ExtractOutsideCompilationForWhileNode()
2113 const string& outside_compilation_attr_name, const string& xla_cluster_name, in ExtractOutsideCompilationForNodesWithAssociatedFunctions()
[all …]
Dextract_outside_compilation_pass_test.cc240 const string &outside_compilation_attr_name, in ExtractOutsideCompilationTest()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc358 const string& outside_compilation_attr_name, const string& xla_func_name, in MoveHeadOutsideCompilationToHost()
999 const string& outside_compilation_attr_name, const string& xla_func_name, in MoveTailOutsideCompilationToHost()
1241 const string& outside_compilation_attr_name, const string& xla_func_name, in ReplaceArgUsedByOutsideCompilationWithPlaceholder()
2883 const string& outside_compilation_attr_name, int* lifted_arg_count, in ProcessHeadTailOutsideCompilation()