Searched refs:else_branch_func (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 700 NameAttrList else_branch_func; in PostprocessLiftedArgsForIf() local
701 TF_RETURN_IF_ERROR(GetNodeAttr(n->def(), "else_branch", &else_branch_func)); in PostprocessLiftedArgsForIf()
703 fld->Find(else_branch_func.name()); in PostprocessLiftedArgsForIf()
719 *else_branch_function_def, AttrSlice(&else_branch_func.attr()), fld, in PostprocessLiftedArgsForIf()
812 absl::StrCat(else_branch_func.name(), "_lifted_arg_")); in PostprocessLiftedArgsForIf()
819 &else_branch_func, n, fld)); in PostprocessLiftedArgsForIf()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tf_passes.td | 453 then_branch = @then_branch_func, else_branch = @else_branch_func, is_stateless = false
464 %1 = call @else_branch_func(%arg1) : (tensor<*xf32>) -> tensor<*xf32>
488 %1 = call @else_branch_func(%arg1) : (tensor<*xf32>) -> tensor<*xf32>
497 then_branch = @then_branch_func, else_branch = @else_branch_func, is_stateless = false
|