Searched refs:max_mem_usage_ (Results 1 – 2 of 2) sorted by relevance
130 max_mem_usage_.resize(id + 1); in Ensure()137 auto max_mem_usage = &max_mem_usage_[id]; in Ensure()262 auto& current_max = max_mem_usage_[id].output_port_mem[output_slot]; in RecordMaxMemorySize()270 max_mem_usage_[id].output_port_shape[output_slot] = tensor_shape; in RecordMaxMemorySize()271 max_mem_usage_[id].output_port_type[output_slot] = dtype; in RecordMaxMemorySize()277 if (id < 0 || static_cast<size_t>(id) >= max_mem_usage_.size() || in MaxMemorySize()278 max_mem_usage_[id].output_port_mem.size() <= static_cast<size_t>(slot)) { in MaxMemorySize()281 return max_mem_usage_[id].output_port_mem[slot]; in MaxMemorySize()287 if (id < 0 || static_cast<size_t>(id) >= max_mem_usage_.size() || in MaxMemoryShape()288 max_mem_usage_[id].output_port_shape.size() <= in MaxMemoryShape()[all …]
218 std::vector<MemUsage> max_mem_usage_; variable