| /external/tensorflow/tensorflow/compiler/jit/ |
| D | compilability_check_util.h | 157 const Node& node, FunctionLibraryRuntime* lib_runtime,
171 const NodeDef& call_def, FunctionLibraryRuntime* lib_runtime,
176 FunctionLibraryRuntime* lib_runtime) const { in IsCompilableNode() argument
179 return IsCompilableNode(node, lib_runtime, &stack_trace); in IsCompilableNode()
185 FunctionLibraryRuntime* lib_runtime) { in IsCompilableCall() argument
188 return IsCompilableCall(call_def, lib_runtime, &stack_trace); in IsCompilableCall()
204 const Node& node, FunctionLibraryRuntime* lib_runtime,
209 const NodeDef& call_def, FunctionLibraryRuntime* lib_runtime,
213 bool IsCompilableIf(const Node& if_node, FunctionLibraryRuntime* lib_runtime,
218 FunctionLibraryRuntime* lib_runtime,
[all …]
|
| D | compilability_check_util.cc | 145 const Node& node, FunctionLibraryRuntime* lib_runtime, in FindUncompilableNodes() argument
162 IsCompilableNode(node, lib_runtime, &stack_trace, in FindUncompilableNodes()
169 const NodeDef& call_def, FunctionLibraryRuntime* lib_runtime, in FindUncompilableNodes() argument
185 IsCompilableCall(call_def, lib_runtime, &stack_trace, in FindUncompilableNodes()
231 const Node& if_node, FunctionLibraryRuntime* lib_runtime, in IsCompilableIf() argument
238 if_node, "then_branch", "if_then", encapsulating_function, lib_runtime, in IsCompilableIf()
243 if_node, "else_branch", "if_else", encapsulating_function, lib_runtime, in IsCompilableIf()
250 const Node& case_node, FunctionLibraryRuntime* lib_runtime, in IsCompilableCase() argument
267 IsCompilableCall(call, lib_runtime, stack_trace, encapsulating_function, in IsCompilableCase()
277 const Node& while_node, FunctionLibraryRuntime* lib_runtime, in IsCompilableWhile() argument
[all …]
|
| D | xla_cluster_util.cc | 394 const Node& n, FunctionLibraryRuntime* lib_runtime) { in GetCallTargetListFromNode() argument
396 *lib_runtime->GetFunctionLibraryDefinition(); in GetCallTargetListFromNode()
421 const Graph& graph, FunctionLibraryRuntime* lib_runtime,
426 FunctionLibraryRuntime* lib_runtime, Direction direction, int depth) { in DoesAnyCalleeHaveRefNodes() argument
449 if (!lib_runtime in DoesAnyCalleeHaveRefNodes()
461 [&] { TF_CHECK_OK(lib_runtime->ReleaseHandle(handle)); }); in DoesAnyCalleeHaveRefNodes()
463 const FunctionBody* fbody = lib_runtime->GetFunctionBody(handle); in DoesAnyCalleeHaveRefNodes()
465 *fbody->graph, lib_runtime, direction, depth + 1, &callee_ref_nodes)); in DoesAnyCalleeHaveRefNodes()
481 const Graph& graph, FunctionLibraryRuntime* lib_runtime, in GetNodesRelatedToRefVariablesInDirection() argument
504 DoesAnyCalleeHaveRefNodes(GetCallTargetListFromNode(*n, lib_runtime), in GetNodesRelatedToRefVariablesInDirection()
[all …]
|
| D | xla_cluster_util.h | 104 const Graph& graph, FunctionLibraryRuntime* lib_runtime);
|
| D | partially_decluster_pass.cc | 300 FunctionLibraryRuntime* lib_runtime = in PartiallyDeclusterGraph() local
304 lib_runtime, IsIntraClusterEdge)); in PartiallyDeclusterGraph()
|
| D | xla_cluster_util_test.cc | 154 FunctionLibraryRuntime* lib_runtime = in GetNodesRelatedToRefVarsSorted() local
158 GetNodesRelatedToRefVariables(*graph, lib_runtime)); in GetNodesRelatedToRefVarsSorted()
|
| D | mark_for_compilation_pass.cc | 1133 FunctionLibraryRuntime* lib_runtime = in FindCompilationCandidates() local
1138 &compile_time_const_nodes, lib_runtime)); in FindCompilationCandidates()
1207 if (!checker.IsCompilableNode(*node, lib_runtime)) { in FindCompilationCandidates()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | resource_util.cc | 48 FunctionLibraryRuntime* lib_runtime,
185 const int call_depth, FunctionLibraryRuntime* lib_runtime, in PropagateThroughCallOp() argument
207 TF_RETURN_IF_ERROR(InstantiateFunctionCall(n.def(), lib_runtime, &handle)); in PropagateThroughCallOp()
209 [&] { TF_CHECK_OK(lib_runtime->ReleaseHandle(handle)); }); in PropagateThroughCallOp()
210 const FunctionBody* fbody = lib_runtime->GetFunctionBody(handle); in PropagateThroughCallOp()
218 lib_runtime, &called_function_source_to_path)); in PropagateThroughCallOp()
256 FunctionLibraryRuntime* lib_runtime, in AnalyzeResourceUsage() argument
300 if (IsFunctionCall(*lib_runtime->GetFunctionLibraryDefinition(), *n)) { in AnalyzeResourceUsage()
302 lib_runtime, &user_to_source, in AnalyzeResourceUsage()
323 const Graph* graph, FunctionLibraryRuntime* lib_runtime, in Analyze() argument
[all …]
|
| D | resource_util.h | 91 const Graph* graph, FunctionLibraryRuntime* lib_runtime,
|
| D | const_analysis_test.cc | 130 FunctionLibraryRuntime* lib_runtime = in TestFunctionCall() local
136 lib_runtime)); in TestFunctionCall()
|
| D | resource_util_test.cc | 60 FunctionLibraryRuntime* lib_runtime = in AnalyzeAndVerify() local
65 TF_EXPECT_OK(ResourceUsageAnalysis::Analyze(graph.get(), lib_runtime, in AnalyzeAndVerify()
|