Searched refs:encapsulating_function (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | compilability_check_util.cc | 233 NameAttrList* encapsulating_function, 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()
252 NameAttrList* encapsulating_function, in IsCompilableCase() argument
267 IsCompilableCall(call, lib_runtime, stack_trace, encapsulating_function, in IsCompilableCase()
279 NameAttrList* encapsulating_function, in IsCompilableWhile() argument
284 while_node, "cond", "while_cond", encapsulating_function, lib_runtime, in IsCompilableWhile()
290 while_node, "body", "while_body", encapsulating_function, lib_runtime, in IsCompilableWhile()
298 const std::string& call_name, NameAttrList* encapsulating_function, in ExtractNodeDefAndCheckCompilability() argument
309 encapsulating_function, uncompilable_nodes); in ExtractNodeDefAndCheckCompilability()
[all …]
|
| D | compilability_check_util.h | 206 NameAttrList* encapsulating_function = nullptr,
211 NameAttrList* encapsulating_function = nullptr,
215 NameAttrList* encapsulating_function,
220 NameAttrList* encapsulating_function,
228 NameAttrList* encapsulating_function,
235 const std::string& call_name, NameAttrList* encapsulating_function,
274 NameAttrList* encapsulating_function,
|