Searched refs:body_func (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_while_op_test.cc | 72 AttrValue body_func; in TEST() local
73 body_func.mutable_func()->set_name("XTimesTwo"); in TEST()
79 .Attr("body", body_func) in TEST()
191 AttrValue body_func; in TEST() local
192 body_func.mutable_func()->set_name("XPlusOneXTimesY"); in TEST()
198 .Attr("body", body_func) in TEST()
305 AttrValue body_func; in TEST() local
306 body_func.mutable_func()->set_name("XTimesTwo"); in TEST()
312 .Attr("body", body_func) in TEST()
|
| D | lower_functional_ops_test.cc | 161 AttrValue body_func; in TEST() local
162 body_func.mutable_func()->set_name("WhileWithIfBody"); in TEST()
167 .Attr("body", body_func) in TEST()
211 NameAttrList body_func; in IfWithWhileThen() local
212 body_func.set_name("XTimesTwo"); in IfWithWhileThen()
230 {"body", body_func}, in IfWithWhileThen()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | function_utils.cc | 222 const FunctionDef* body_func = in IsNodeStateful() local
226 (body_func != nullptr && in IsNodeStateful()
227 !IsFunctionStateful(library, *body_func, skip_assert))) { in IsNodeStateful()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla_util.cc | 256 const FunctionDef* body_func = lookup_fld->Find(body_attr.name()); in PropagateConstIntoWhileNode() local
257 if (!body_func) { in PropagateConstIntoWhileNode()
273 if (i >= body_func->signature().output_arg_size()) { in PropagateConstIntoWhileNode()
276 const OpDef_ArgDef& output_arg = body_func->signature().output_arg(i); in PropagateConstIntoWhileNode()
277 auto output_arg_input = body_func->ret().find(output_arg.name()); in PropagateConstIntoWhileNode()
278 if (output_arg_input == body_func->ret().end()) { in PropagateConstIntoWhileNode()
283 const OpDef_ArgDef& input_arg = body_func->signature().input_arg(i); in PropagateConstIntoWhileNode()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | convert_to_constants.py | 499 body_func = while_node.attr["body"].func.name
504 type=function_data[body_func]["types"][idx])
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | captured_function.cc | 394 const FunctionDef* body_func = in IsNodeStateful() local
399 if (body_func != nullptr) { in IsNodeStateful()
400 TF_RETURN_IF_ERROR(IsFunctionStateful(library, *body_func)); in IsNodeStateful()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 574 NameAttrList body_func; in PostprocessLiftedArgsForWhile() local
575 TF_RETURN_IF_ERROR(GetNodeAttr(n->def(), "body", &body_func)); in PostprocessLiftedArgsForWhile()
576 const FunctionDef* body_function_def = fld->Find(body_func.name()); in PostprocessLiftedArgsForWhile()
586 AttrSlice(&body_func.attr()), fld, in PostprocessLiftedArgsForWhile()
636 *body_function_body->graph, body_func.name(), HostGraphControlRetMapping, in PostprocessLiftedArgsForWhile()
639 fld->ReplaceFunction(body_func.name(), rewritten_body_function_def)); in PostprocessLiftedArgsForWhile()
|