Searched refs:output_port_shape (Results 1 – 2 of 2) sorted by relevance
142 DCHECK_EQ(max_mem_usage->output_port_shape.size(), perslot->size()); in Ensure()148 max_mem_usage->output_port_shape.resize(num_outputs, unknown_shape_); in Ensure()270 max_mem_usage_[id].output_port_shape[output_slot] = tensor_shape; in RecordMaxMemorySize()288 max_mem_usage_[id].output_port_shape.size() <= in MaxMemoryShape()292 return max_mem_usage_[id].output_port_shape[slot]; in MaxMemoryShape()
215 gtl::InlinedVector<TensorShapeProto, 2> output_port_shape; member