Searched refs:encapsulating_function (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | compilability_check_util.cc | 216 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 …]
|
| D | compilability_check_util.h | 192 NameAttrList* encapsulating_function = nullptr,
197 NameAttrList* encapsulating_function = nullptr,
201 NameAttrList* encapsulating_function,
206 NameAttrList* encapsulating_function,
214 NameAttrList* encapsulating_function,
221 const std::string& call_name, NameAttrList* encapsulating_function,
260 NameAttrList* encapsulating_function,
|