Home
last modified time | relevance | path

Searched refs:encapsulating_function (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dcompilability_check_util.cc216 NameAttrList* encapsulating_function, in IsCompilableIf() argument
221 if_node, "then_branch", "if_then", encapsulating_function, lib_runtime, in IsCompilableIf()
226 if_node, "else_branch", "if_else", encapsulating_function, lib_runtime, in IsCompilableIf()
235 NameAttrList* encapsulating_function, in IsCompilableCase() argument
250 IsCompilableCall(call, lib_runtime, stack_trace, encapsulating_function, in IsCompilableCase()
262 NameAttrList* encapsulating_function, in IsCompilableWhile() argument
267 while_node, "cond", "while_cond", encapsulating_function, lib_runtime, in IsCompilableWhile()
273 while_node, "body", "while_body", encapsulating_function, lib_runtime, in IsCompilableWhile()
281 const std::string& call_name, NameAttrList* encapsulating_function, in ExtractNodeDefAndCheckCompilability() argument
292 encapsulating_function, uncompilable_nodes); in ExtractNodeDefAndCheckCompilability()
[all …]
H A Dcompilability_check_util.h202 NameAttrList* encapsulating_function = nullptr,
207 NameAttrList* encapsulating_function = nullptr,
211 NameAttrList* encapsulating_function,
216 NameAttrList* encapsulating_function,
224 NameAttrList* encapsulating_function,
231 const std::string& call_name, NameAttrList* encapsulating_function,
270 NameAttrList* encapsulating_function,