Searched refs:get_function_body_fn (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | rearrange_function_argument.cc | 278 get_function_body_fn, in MaybeRewriteWhileNode() argument
311 TF_RETURN_IF_ERROR(get_function_body_fn(attr_value, &fbody)); in MaybeRewriteWhileNode()
382 get_function_body_fn, in MaybeRewriteIfNode() argument
437 TF_RETURN_IF_ERROR(get_function_body_fn(f, &fbody)); in MaybeRewriteIfNode()
512 get_function_body_fn, in RearrangeFunctionArguments() argument
526 TF_RETURN_IF_ERROR(get_function_body_fn(func_name_attrs, &fbody)); in RearrangeFunctionArguments()
539 TF_RETURN_IF_ERROR(MaybeRewriteWhileNode(get_function_body_fn, g, n, fld, in RearrangeFunctionArguments()
543 TF_RETURN_IF_ERROR(MaybeRewriteIfNode(get_function_body_fn, g, n, fld, in RearrangeFunctionArguments()
|
| D | rearrange_function_argument.h | 36 get_function_body_fn,
|