Searched refs:body_function (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | const_analysis.cc | 140 NameAttrList body_function; in GetCompileTimeConstInputs() local
141 TF_RETURN_IF_ERROR(GetNodeAttr(node->def(), "body", &body_function)); in GetCompileTimeConstInputs()
147 body_function.name(), AttrSlice(&body_function.attr()), &body_handle)); in GetCompileTimeConstInputs()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | loops.cc | 27 const WhileLoopHelperBodyFunction& body_function, in WhileLoopHelper() argument
69 body_function(unpack_tuple(parameter, arity, body_builder.get()), in WhileLoopHelper()
84 const ForEachIndexBodyFunction& body_function, in ForEachIndex() argument
105 body_function(iteration, values, body_builder)); in ForEachIndex()
|
| D | loops.h | 51 const WhileLoopHelperBodyFunction& body_function,
66 const ForEachIndexBodyFunction& body_function,
|