Searched refs:then_func (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_functional_ops_test.cc | 89 NameAttrList then_func; in WhileWithIfBody() local
90 then_func.set_name("XTimesTwo"); in WhileWithIfBody()
110 {{"then_branch", then_func}, in WhileWithIfBody()
262 AttrValue then_func; in TEST() local
263 then_func.mutable_func()->set_name("IfWithWhileThen"); in TEST()
270 .Attr("then_branch", then_func) in TEST()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | optimize_functional_ops.cc | 86 FuncOp then_func = op.then_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 | 207 const FunctionDef* then_func = in IsNodeStateful() local
211 if ((then_func != nullptr && in IsNodeStateful()
212 !IsFunctionStateful(library, *then_func, skip_assert)) && in IsNodeStateful()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | stack_ops_decomposition.cc | 232 auto then_func = if_op.then_function(); in HandleIfOp() local
242 ModifyFunctionSignature(then_func, &then_map, find_arg_stack_type); in HandleIfOp()
245 if (failed(DecomposeStackOpsInternal(&then_func.front(), module, &then_map, in HandleIfOp()
259 if_op.getLoc(), then_func.getType().getResults(), new_if_operands, in HandleIfOp()
266 FindAliasedInput(then_func, result.getResultNumber()); in HandleIfOp()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | captured_function.cc | 416 const FunctionDef* then_func = in IsNodeStateful() local
420 if (then_func != nullptr) { in IsNodeStateful()
421 TF_RETURN_IF_ERROR(IsFunctionStateful(library, *then_func)); in IsNodeStateful()
|