Searched refs:then_branch_func (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 694 NameAttrList then_branch_func; in PostprocessLiftedArgsForIf() local
695 TF_RETURN_IF_ERROR(GetNodeAttr(n->def(), "then_branch", &then_branch_func)); in PostprocessLiftedArgsForIf()
697 fld->Find(then_branch_func.name()); in PostprocessLiftedArgsForIf()
714 *then_branch_function_def, AttrSlice(&then_branch_func.attr()), fld, in PostprocessLiftedArgsForIf()
802 absl::StrCat(then_branch_func.name(), "_lifted_arg_")); in PostprocessLiftedArgsForIf()
809 &then_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
461 %1 = call @then_branch_func(%arg1) : (tensor<*xf32>) -> tensor<*xf32>
485 %1 = call @then_branch_func(%arg1) : (tensor<*xf32>) -> tensor<*xf32>
497 then_branch = @then_branch_func, else_branch = @else_branch_func, is_stateless = false
|