Searched refs:_get_tensor_index_in_iterable (Results 1 – 1 of 1) sorted by relevance
83 def _get_tensor_index_in_iterable(iterable, t): function101 idx = _get_tensor_index_in_iterable(body_grad_graph.structured_outputs,271 structured_idx = _get_tensor_index_in_iterable(graph.structured_outputs,276 flat_idx = _get_tensor_index_in_iterable(