Searched refs:body_fdef (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla_util_test.cc | 352 FunctionDef body_fdef; in TEST() local
353 TF_ASSERT_OK(GraphToFunctionDef(graph, "body", &body_fdef)); in TEST()
354 TF_ASSERT_OK(fld.AddFunctionDef(body_fdef)); in TEST()
385 FunctionDef body_fdef; in TEST() local
386 TF_ASSERT_OK(GraphToFunctionDef(graph, "body", &body_fdef)); in TEST()
387 TF_ASSERT_OK(fld.AddFunctionDef(body_fdef)); in TEST()
|
| D | functionalize_while.cc | 426 FunctionDef body_fdef; in FunctionalizeLoop() local
428 GraphToFunctionDef(*body_graph, body_name.name(), &body_fdef)); in FunctionalizeLoop()
431 TF_RETURN_IF_ERROR(library->AddFunctionDef(body_fdef)); in FunctionalizeLoop()
|
| D | functionalize_cond.cc | 768 FunctionDef body_fdef; in BuildIfNode() local
770 body_name.name(), &body_fdef)); in BuildIfNode()
771 TF_RETURN_IF_ERROR(library->AddFunctionDef(body_fdef)); in BuildIfNode()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 1798 const FunctionDef* body_fdef = fld->Find(body_func.name()); in FindArgsToLiftForWhileNode() local
1799 if (body_fdef == nullptr) { in FindArgsToLiftForWhileNode()
1805 *body_fdef, AttrSlice(&body_func.attr()), fld, &body_fbody)); in FindArgsToLiftForWhileNode()
|