Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dcontrol_flow_util_v2.py188 def _extract_input_index(function_attribute_name): function
207 tensor_name = node_def.input[_extract_input_index("f")]
209 input_index = _extract_input_index("then_branch")
210 if input_index != _extract_input_index("else_branch"):