Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/graph/
Dcostmodel.cc137 auto max_mem_usage = &max_mem_usage_[id]; in Ensure() local
141 DCHECK_EQ(max_mem_usage->output_port_mem.size(), perslot->size()); in Ensure()
142 DCHECK_EQ(max_mem_usage->output_port_shape.size(), perslot->size()); in Ensure()
143 DCHECK_EQ(max_mem_usage->output_port_type.size(), perslot->size()); in Ensure()
147 max_mem_usage->output_port_mem.resize(num_outputs, Bytes(-1)); in Ensure()
148 max_mem_usage->output_port_shape.resize(num_outputs, unknown_shape_); in Ensure()
149 max_mem_usage->output_port_type.resize(num_outputs, DT_INVALID); in Ensure()