Searched refs:alloc_id (Results 1 – 6 of 6) sorted by relevance
327 for (int64 alloc_id : memory_stats.persistent_tensor_alloc_ids()) { in RecordMemoryStats() local328 if (alloc_id > 0) { in RecordMemoryStats()329 persistent_alloc_ids_.insert(alloc_id); in RecordMemoryStats()350 int64 alloc_id) { in RecordAllocationId() argument354 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() local515 if (input_alloc_id == alloc_id) { in AddToCostGraphDef()[all …]
151 void RecordAllocationId(const Node* node, int output_slot, int64 alloc_id);156 bool IsPersistentTensor(const Node* node, int64 alloc_id) const;
197 for (const auto& alloc_id : context->persistent_alloc_ids()) { in UpdateStats() local198 ms->mutable_persistent_tensor_alloc_ids()->Add(alloc_id); in UpdateStats()
775 int64 alloc_id = a->AllocationId(t->tensor_data().data()); in allocate_persistent() local776 record_persistent_memory_allocation(alloc_size, alloc_id); in allocate_persistent()901 int64 alloc_id) { in record_persistent_memory_allocation() argument904 if (alloc_id >= 0) { in record_persistent_memory_allocation()908 persistent_alloc_ids_->push_back(alloc_id); in record_persistent_memory_allocation()
1197 void record_persistent_memory_allocation(int64 size, int64 alloc_id = -1)
147 for (const auto& alloc_id : ctx->persistent_alloc_ids()) { in SetMemory() local148 ms->mutable_persistent_tensor_alloc_ids()->Add(alloc_id); in SetMemory()