Searched refs:MarkLiveAtIndex (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_liveness_analysis.cc | 66 void MarkLiveAtIndex(const HloInstruction* instruction, in MarkLiveAtIndex() function
131 MarkLiveAtIndex(instruction->operand(operand_index), {}, live_index_map, in PropagateLivenessThroughTuple()
138 MarkLiveAtIndex(instruction->operand(operand_index), operand_shape_index, in PropagateLivenessThroughTuple()
153 MarkLiveAtIndex(instruction->operand(0), {}, live_index_map, worklist, in PropagateLivenessThroughGTE()
160 MarkLiveAtIndex(instruction->operand(0), operand_shape_index, in PropagateLivenessThroughGTE()
178 MarkLiveAtIndex(instruction->while_body()->root_instruction(), shape_index, in PropagateLivenessThroughWhile()
181 MarkLiveAtIndex(instruction->operand(0), shape_index, live_index_map, in PropagateLivenessThroughWhile()
186 MarkLiveAtIndex(instruction->while_condition()->root_instruction(), {}, in PropagateLivenessThroughWhile()
209 MarkLiveAtIndex(xla_while, shape_index, live_index_map, worklist, in PropagateLivenessToParameterCallers()
212 MarkLiveAtIndex(xla_while->while_body()->root_instruction(), in PropagateLivenessToParameterCallers()
[all …]
|