Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_simplifier.cc389 int index_to_investigate = 0; in TryRemoveRepeatedWhileTupleIndices() local
409 while (index_to_investigate < while_init->shape().tuple_shapes_size()) { in TryRemoveRepeatedWhileTupleIndices()
422 VLOG(2) << "Iterating " << index_to_investigate; in TryRemoveRepeatedWhileTupleIndices()
425 auto* pivot_init_elem = while_init->operand(index_to_investigate); in TryRemoveRepeatedWhileTupleIndices()
426 auto* pivot_body_elem = while_body_root->operand(index_to_investigate); in TryRemoveRepeatedWhileTupleIndices()
429 if (pivot_body_elem->tuple_index() != index_to_investigate) { in TryRemoveRepeatedWhileTupleIndices()
432 << index_to_investigate; in TryRemoveRepeatedWhileTupleIndices()
433 index_to_investigate++; in TryRemoveRepeatedWhileTupleIndices()
437 index_to_investigate++; in TryRemoveRepeatedWhileTupleIndices()
442 for (int64 i = index_to_investigate + 1; in TryRemoveRepeatedWhileTupleIndices()
[all …]