| /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()
194 AttrValue body_func; in TEST() local
195 body_func.mutable_func()->set_name("XTimesTwo"); in TEST()
201 .Attr("body", body_func) in TEST()
299 AttrValue body_func; in TEST() local
300 body_func.mutable_func()->set_name("XTimesTwo"); in TEST()
308 .Attr("body", body_func) in TEST()
423 AttrValue body_func; in TEST() local
[all …]
|
| D | lower_functional_ops_test.cc | 160 AttrValue body_func; in TEST() local
161 body_func.mutable_func()->set_name("WhileWithIfBody"); in TEST()
166 .Attr("body", body_func) in TEST()
210 NameAttrList body_func; in IfWithWhileThen() local
211 body_func.set_name("XTimesTwo"); in IfWithWhileThen()
229 {"body", body_func}, in IfWithWhileThen()
|
| D | executor_test.cc | 624 AttrValue body_func; in BM_WhileLoopHelper() local
625 body_func.mutable_func()->set_name("XPlusOne"); in BM_WhileLoopHelper()
631 .Attr("body", body_func) in BM_WhileLoopHelper()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | py_context_manager_pybind.cc | 27 py::handle body_func) { in TestPyContextManager() argument
36 PyObject_CallFunctionObjArgs(body_func.ptr(), cm.var(), nullptr)); in TestPyContextManager()
|
| /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 | 260 const FunctionDef* body_func = lookup_fld->Find(body_attr.name()); in PropagateConstIntoWhileNode() local
261 if (!body_func) { in PropagateConstIntoWhileNode()
277 if (i >= body_func->signature().output_arg_size()) { in PropagateConstIntoWhileNode()
280 const OpDef_ArgDef& output_arg = body_func->signature().output_arg(i); in PropagateConstIntoWhileNode()
281 auto output_arg_input = body_func->ret().find(output_arg.name()); in PropagateConstIntoWhileNode()
282 if (output_arg_input == body_func->ret().end()) { in PropagateConstIntoWhileNode()
287 const OpDef_ArgDef& input_arg = body_func->signature().input_arg(i); in PropagateConstIntoWhileNode()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | captured_function.cc | 432 const FunctionDef* body_func = in IsNodeStateful() local
437 if (body_func != nullptr) { in IsNodeStateful()
438 TF_RETURN_IF_ERROR(IsFunctionStateful(library, *body_func)); in IsNodeStateful()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 590 NameAttrList body_func; in PostprocessLiftedArgsForWhile() local
591 TF_RETURN_IF_ERROR(GetNodeAttr(n->def(), "body", &body_func)); in PostprocessLiftedArgsForWhile()
592 const FunctionDef* body_function_def = fld->Find(body_func.name()); in PostprocessLiftedArgsForWhile()
602 AttrSlice(&body_func.attr()), fld, in PostprocessLiftedArgsForWhile()
651 fld->UniqueFunctionName(absl::StrCat(body_func.name(), "_lifted_arg_")); in PostprocessLiftedArgsForWhile()
658 &body_func, n, fld)); in PostprocessLiftedArgsForWhile()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 1796 NameAttrList body_func; in FindArgsToLiftForWhileNode() local
1797 TF_RETURN_IF_ERROR(GetNodeAttr(while_node->def(), "body", &body_func)); in FindArgsToLiftForWhileNode()
1798 const FunctionDef* body_fdef = fld->Find(body_func.name()); in FindArgsToLiftForWhileNode()
1800 return errors::Internal("Cannot find body function ", body_func.name(), in FindArgsToLiftForWhileNode()
1805 *body_fdef, AttrSlice(&body_func.attr()), fld, &body_fbody)); in FindArgsToLiftForWhileNode()
|