Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_rematerialization.cc547 Status AddCompressInstructions(Item* original_item, Item* compressed_item,
1012 Item* compressed_item, in AddCompressInstructions() argument
1017 memory_usage_ += size_function_(compressed_item->instruction->shape()); in AddCompressInstructions()
1034 original_buffer.users.push_back(ItemUse{compressed_item, 0, absl::nullopt}); in AddCompressInstructions()
1040 NewBuffer(compressed_item, compressed_item->instruction->shape(), in AddCompressInstructions()
1044 compressed_item->buffers_used = original_item->buffers_output; in AddCompressInstructions()
1045 compressed_item->buffers_output = {compressed_buffer.id}; in AddCompressInstructions()
1046 compressed_item->buffers_defined.push_back(compressed_buffer.id); in AddCompressInstructions()
1053 uncompressed_item->buffers_used = {compressed_item->buffers_output[0]}; in AddCompressInstructions()
1700 Item* compressed_item = instruction_list->CreateItem(compressed); in CompressInstruction() local
[all …]