Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/graph/
Dcostmodel.cc327 for (int64 alloc_id : memory_stats.persistent_tensor_alloc_ids()) { in RecordMemoryStats() local
328 if (alloc_id > 0) { in RecordMemoryStats()
329 persistent_alloc_ids_.insert(alloc_id); in RecordMemoryStats()
350 int64 alloc_id) { in RecordAllocationId() argument
354 output_port_alloc_ids_[id][output_slot] = alloc_id; in RecordAllocationId()
366 bool CostModel::IsPersistentTensor(const Node* node, int64 alloc_id) const { in IsPersistentTensor()
367 if (persistent_alloc_ids_.count(alloc_id) > 0) { in IsPersistentTensor()
375 .count(alloc_id); in IsPersistentTensor()
511 int64 alloc_id = AllocationId(n, i); in AddToCostGraphDef() local
515 if (input_alloc_id == alloc_id) { in AddToCostGraphDef()
[all …]
Dcostmodel.h151 void RecordAllocationId(const Node* node, int output_slot, int64 alloc_id);
156 bool IsPersistentTensor(const Node* node, int64 alloc_id) const;
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.cc197 for (const auto& alloc_id : context->persistent_alloc_ids()) { in UpdateStats() local
198 ms->mutable_persistent_tensor_alloc_ids()->Add(alloc_id); in UpdateStats()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.cc775 int64 alloc_id = a->AllocationId(t->tensor_data().data()); in allocate_persistent() local
776 record_persistent_memory_allocation(alloc_size, alloc_id); in allocate_persistent()
901 int64 alloc_id) { in record_persistent_memory_allocation() argument
904 if (alloc_id >= 0) { in record_persistent_memory_allocation()
908 persistent_alloc_ids_->push_back(alloc_id); in record_persistent_memory_allocation()
Dop_kernel.h1197 void record_persistent_memory_allocation(int64 size, int64 alloc_id = -1)
/external/tensorflow/tensorflow/core/common_runtime/
Dstep_stats_collector.cc147 for (const auto& alloc_id : ctx->persistent_alloc_ids()) { in SetMemory() local
148 ms->mutable_persistent_tensor_alloc_ids()->Add(alloc_id); in SetMemory()