Home
last modified time | relevance | path

Searched refs:outside_compilation_attr_name (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_util.cc74 Graph* g, const string& outside_compilation_attr_name) { in PreprocessControlEdgesBetweenOutsideCompilations() argument
83 GetStringAttr(*e->src(), outside_compilation_attr_name); in PreprocessControlEdgesBetweenOutsideCompilations()
85 GetStringAttr(*e->dst(), outside_compilation_attr_name); in PreprocessControlEdgesBetweenOutsideCompilations()
114 Graph* g, const string& outside_compilation_attr_name) { in PreprocessDataEdgesBetweenOutsideCompilations() argument
128 GetStringAttr(*e->src(), outside_compilation_attr_name); in PreprocessDataEdgesBetweenOutsideCompilations()
130 GetStringAttr(*e->dst(), outside_compilation_attr_name); in PreprocessDataEdgesBetweenOutsideCompilations()
161 outside_compilation_attr_name, in PreprocessDataEdgesBetweenOutsideCompilations()
163 placeholder_builder.Attr(outside_compilation_attr_name, in PreprocessDataEdgesBetweenOutsideCompilations()
200 Graph* g, const string& outside_compilation_attr_name) { in PostprocessDataEdgesBetweenOutsideCompilations() argument
276 Graph* g, const string& outside_compilation_attr_name) { in PostprocessControlEdgesBetweenOutsideCompilations() argument
[all …]
Dextract_outside_compilation_pass.h48 const string& outside_compilation_attr_name, in RewriteOutsideCompilationSubgraphFn() argument
51 outside_compilation_attr_name_(outside_compilation_attr_name), in RewriteOutsideCompilationSubgraphFn()
91 const string& outside_compilation_attr_name, const string& xla_cluster_name,
104 const string& outside_compilation_attr_name,
Dencapsulate_util.h123 const Graph* g, const string& outside_compilation_attr_name);
139 Graph* g, const string& outside_compilation_attr_name);
153 Graph* g, const string& outside_compilation_attr_name);
Dextract_outside_compilation_pass.cc982 const string& xla_cluster_name, const string& outside_compilation_attr_name, in ConstructHostGraph() argument
1103 host_graph->get(), outside_compilation_attr_name)); in ConstructHostGraph()
1393 const string& outside_compilation_attr_name, const string& xla_cluster_name, in BuildHostGraphForIfNode() argument
1415 recv_pred_builder.Attr(outside_compilation_attr_name, in BuildHostGraphForIfNode()
1447 if_builder.Attr(outside_compilation_attr_name, outside_compilation_name); in BuildHostGraphForIfNode()
1553 const string& xla_cluster_name, const string& outside_compilation_attr_name, in RewriteHostWhileLoopCond() argument
1590 recv_pred_builder.Attr(outside_compilation_attr_name, in RewriteHostWhileLoopCond()
1629 const string& xla_cluster_name, const string& outside_compilation_attr_name, in RewriteHostWhileLoopBody() argument
1688 const string& outside_compilation_attr_name, const string& xla_cluster_name, in BuildHostGraphForWhileNode() argument
1703 xla_cluster_attr_name, xla_cluster_name, outside_compilation_attr_name, in BuildHostGraphForWhileNode()
[all …]
Dextract_outside_compilation_pass_test.cc240 const string &outside_compilation_attr_name, in ExtractOutsideCompilationTest() argument
254 xla_cluster_attr_name, outside_compilation_attr_name, xla_cluster_name, in ExtractOutsideCompilationTest()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.h66 const string& outside_compilation_attr_name, int* lifted_arg_count,
Dencapsulate_tpu_computations_pass.cc358 const string& outside_compilation_attr_name, const string& xla_func_name, in MoveHeadOutsideCompilationToHost() argument
374 (!HasNodeAttr(e->src()->def(), outside_compilation_attr_name) || in MoveHeadOutsideCompilationToHost()
380 if (HasNodeAttr(n->def(), outside_compilation_attr_name) && in MoveHeadOutsideCompilationToHost()
399 HasNodeAttr(e->dst()->def(), outside_compilation_attr_name) && in MoveHeadOutsideCompilationToHost()
999 const string& outside_compilation_attr_name, const string& xla_func_name, in MoveTailOutsideCompilationToHost() argument
1015 (!HasNodeAttr(e->dst()->def(), outside_compilation_attr_name) || in MoveTailOutsideCompilationToHost()
1021 if (HasNodeAttr(n->def(), outside_compilation_attr_name) && in MoveTailOutsideCompilationToHost()
1241 const string& outside_compilation_attr_name, const string& xla_func_name, in ReplaceArgUsedByOutsideCompilationWithPlaceholder() argument
2883 const string& outside_compilation_attr_name, int* lifted_arg_count, in ProcessHeadTailOutsideCompilation() argument
2919 outside_compilation_attr_name, iter.second.func_name_attrs.name(), in ProcessHeadTailOutsideCompilation()
[all …]