Home
last modified time | relevance | path

Searched refs:body_func (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dlower_while_op_test.cc62 AttrValue body_func; in TEST() local
63 body_func.mutable_func()->set_name("XTimesTwo"); in TEST()
68 .Attr("body", body_func) in TEST()
165 AttrValue body_func; in TEST() local
166 body_func.mutable_func()->set_name("XPlusOneXTimesY"); in TEST()
171 .Attr("body", body_func) in TEST()
267 AttrValue body_func; in TEST() local
268 body_func.mutable_func()->set_name("XTimesTwo"); in TEST()
273 .Attr("body", body_func) in TEST()
Dlower_if_while_test.cc148 AttrValue body_func; in TEST() local
149 body_func.mutable_func()->set_name("WhileWithIfBody"); in TEST()
154 .Attr("body", body_func) in TEST()
198 NameAttrList body_func; in IfWithWhileThen() local
199 body_func.set_name("XTimesTwo"); in IfWithWhileThen()
217 {"body", body_func}, in IfWithWhileThen()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dfunction_utils.cc213 const FunctionDef* body_func = in IsNodeStateful() local
217 (body_func != nullptr && in IsNodeStateful()
218 !IsFunctionStateful(library, *body_func, skip_assert))) { in IsNodeStateful()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util.cc261 const FunctionDef* body_func = lookup_fld->Find(body_attr.name()); in PropagateConstIntoWhileNode() local
262 if (!body_func) { in PropagateConstIntoWhileNode()
278 if (i >= body_func->signature().output_arg_size()) { in PropagateConstIntoWhileNode()
281 const OpDef_ArgDef& output_arg = body_func->signature().output_arg(i); in PropagateConstIntoWhileNode()
282 auto output_arg_input = body_func->ret().find(output_arg.name()); in PropagateConstIntoWhileNode()
283 if (output_arg_input == body_func->ret().end()) { in PropagateConstIntoWhileNode()
288 const OpDef_ArgDef& input_arg = body_func->signature().input_arg(i); in PropagateConstIntoWhileNode()
/external/tensorflow/tensorflow/compiler/jit/
Dmark_for_compilation_pass.cc203 const string body_func = name_attr->name(); in IsCompilableWhile() local
205 call.set_op(body_func); in IsCompilableWhile()
210 << ": can't compile loop body: " << body_func; in IsCompilableWhile()