Home
last modified time | relevance | path

Searched refs:branch_inputs (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_control_flow.cc287 llvm::SmallVector<Value, 4> branch_inputs; in LowerCaseRegion() local
288 branch_inputs.reserve(op.branches().size()); in LowerCaseRegion()
293 branch_inputs.emplace_back(branch_input); in LowerCaseRegion()
303 branch_inputs, branch_inputs.size()); in LowerCaseRegion()
/external/tensorflow/tensorflow/python/ops/
Dcond_v2.py509 def _make_inputs_match(branch_graphs, branch_inputs): argument
526 assert len(branch_graphs) == len(branch_inputs)
529 for branch_in in branch_inputs:
536 for branch_graph, branch_in in zip(branch_graphs, branch_inputs):
1121 branch_inputs, argument
1150 case_inputs = _make_inputs_match(branch_graphs, branch_inputs)