Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_rematerialization.cc547 Status AddCompressInstructions(Item* original_item, Item* compressed_item,
555 Status AddRematerializedInstruction(Item* original_item, Item* remat_item,
1011 Status MemoryUsageTracker::AddCompressInstructions(Item* original_item, in AddCompressInstructions() argument
1015 memory_usage_ -= size_function_(original_item->instruction->shape()); in AddCompressInstructions()
1021 CHECK_EQ(original_item->buffers_output.size(), 1); in AddCompressInstructions()
1022 BufferId original_buffer_id = original_item->buffers_output[0]; in AddCompressInstructions()
1044 compressed_item->buffers_used = original_item->buffers_output; in AddCompressInstructions()
1067 Item* original_item, Item* remat_item, absl::Span<Item*> indirect_users) { in AddRematerializedInstruction() argument
1069 << original_item->instruction->name() in AddRematerializedInstruction()
1073 TF_RET_CHECK(original_item->placed) << original_item->instruction->name(); in AddRematerializedInstruction()
[all …]
/external/tensorflow/tensorflow/python/framework/
Dops.py6071 def _assert_same_graph(original_item, item): argument
6081 original_graph = getattr(original_item, "graph", None)
6086 (item, original_item, graph, original_graph))
/external/tensorflow/tensorflow/python/keras/
Dbackend.py610 def _assert_same_graph(original_item, item): argument
620 original_graph = getattr(original_item, 'graph', None)
625 (item, original_item, graph, original_graph))