Searched refs:else_func (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_functional_ops_test.cc | 91 NameAttrList else_func; in WhileWithIfBody() local
92 else_func.set_name("XTimesFour"); in WhileWithIfBody()
111 {"else_branch", else_func}, in WhileWithIfBody()
264 AttrValue else_func; in TEST() local
265 else_func.mutable_func()->set_name("XTimesTwo"); in TEST()
271 .Attr("else_branch", else_func) in TEST()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | optimize_functional_ops.cc | 87 FuncOp else_func = op.else_function(); in matchAndRewrite() local
94 (IsSideEffectFree(then_func) && IsSideEffectFree(else_func)))) { in matchAndRewrite()
111 FuncOp func = cond_value ? then_func : else_func; in matchAndRewrite()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | function_utils.cc | 209 const FunctionDef* else_func = in IsNodeStateful() local
213 (else_func != nullptr && in IsNodeStateful()
214 !IsFunctionStateful(library, *else_func, skip_assert))) { in IsNodeStateful()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | stack_ops_decomposition.cc | 233 auto else_func = if_op.else_function(); in HandleIfOp() local
243 ModifyFunctionSignature(else_func, &else_map, find_arg_stack_type); in HandleIfOp()
247 failed(DecomposeStackOpsInternal(&else_func.front(), module, &else_map, in HandleIfOp()
268 FindAliasedInput(else_func, result.getResultNumber()); in HandleIfOp()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | captured_function.cc | 418 const FunctionDef* else_func = in IsNodeStateful() local
423 if (else_func != nullptr) { in IsNodeStateful()
424 TF_RETURN_IF_ERROR(IsFunctionStateful(library, *else_func)); in IsNodeStateful()
|