Searched refs:max_mem_usage_ (Results 1 – 2 of 2) sorted by relevance
132 max_mem_usage_.resize(id + 1); in Ensure()139 auto max_mem_usage = &max_mem_usage_[id]; in Ensure()264 auto& current_max = max_mem_usage_[id].output_port_mem[output_slot]; in RecordMaxMemorySize()272 max_mem_usage_[id].output_port_shape[output_slot] = tensor_shape; in RecordMaxMemorySize()273 max_mem_usage_[id].output_port_type[output_slot] = dtype; in RecordMaxMemorySize()279 if (id < 0 || static_cast<size_t>(id) >= max_mem_usage_.size() || in MaxMemorySize()280 max_mem_usage_[id].output_port_mem.size() <= static_cast<size_t>(slot)) { in MaxMemorySize()283 return max_mem_usage_[id].output_port_mem[slot]; in MaxMemorySize()289 if (id < 0 || static_cast<size_t>(id) >= max_mem_usage_.size() || in MaxMemoryShape()290 max_mem_usage_[id].output_port_shape.size() <= in MaxMemoryShape()[all …]
226 std::vector<MemUsage> max_mem_usage_; variable