Searched refs:alloc_id (Results 1 – 5 of 5) 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;
155 for (const auto& alloc_id : ctx->persistent_alloc_ids()) { in SetMemory() local156 ms->mutable_persistent_tensor_alloc_ids()->Add(alloc_id); in SetMemory()
885 int64 alloc_id = a->AllocationId(t->tensor_data().data()); in allocate_persistent() local886 record_persistent_memory_allocation(alloc_size, alloc_id); in allocate_persistent()1058 int64 alloc_id) { in record_persistent_memory_allocation() argument1062 if (alloc_id >= 0) { in record_persistent_memory_allocation()1063 tracking_state_->persistent_alloc_ids.push_back(alloc_id); in record_persistent_memory_allocation()
1249 void record_persistent_memory_allocation(int64 size, int64 alloc_id = -1)