Searched refs:add_gte (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_loop_simplifier.cc | 1058 auto add_gte = [&](HloInstruction* src, int64 idx) { in TryMergeInductionVariables() local
1084 tuple_elems.push_back(add_gte(instr, i)); in TryMergeInductionVariables()
1088 elem_shape, HloOpcode::kAdd, add_gte(instr, i), in TryMergeInductionVariables()
1090 add_gte(instr, *trip_counter), in TryMergeInductionVariables()
1109 add_gte((induction_vars.count(i) ? loop_body_param : old_root), i)); in TryMergeInductionVariables()
1116 add_gte(loop_body_param, *trip_counter), in TryMergeInductionVariables()
1133 tuple_elems.push_back(add_gte(init, i)); in TryMergeInductionVariables()
|