Searched refs:remat_item (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_rematerialization.cc | 555 Status AddRematerializedInstruction(Item* original_item, Item* remat_item,
666 Buffer& RematerializeBuffer(const Buffer& original_buffer, Item* remat_item, in RematerializeBuffer() argument
675 return NewBuffer(remat_item, original_buffer.shape, original_buffer.index, in RematerializeBuffer()
1067 Item* original_item, Item* remat_item, absl::Span<Item*> indirect_users) { in AddRematerializedInstruction() argument
1070 << ", remat_instruction = " << remat_item->instruction->name(); in AddRematerializedInstruction()
1074 TF_RET_CHECK(!remat_item->placed) << remat_item->instruction->name(); in AddRematerializedInstruction()
1077 remat_item->buffers_used = original_item->buffers_used; in AddRematerializedInstruction()
1094 buffer.users.push_back(ItemUse{remat_item, u.operand_number, u.index}); in AddRematerializedInstruction()
1134 RematerializeBuffer(old_buffer, remat_item, std::move(unplaced_users)); in AddRematerializedInstruction()
1136 remat_item->buffers_defined.push_back(new_buffer.id); in AddRematerializedInstruction()
[all …]
|