Searched refs:output_port_shape (Results 1 – 2 of 2) sorted by relevance
144 DCHECK_EQ(max_mem_usage->output_port_shape.size(), perslot->size()); in Ensure()150 max_mem_usage->output_port_shape.resize(num_outputs, unknown_shape_); in Ensure()272 max_mem_usage_[id].output_port_shape[output_slot] = tensor_shape; in RecordMaxMemorySize()290 max_mem_usage_[id].output_port_shape.size() <= in MaxMemoryShape()294 return max_mem_usage_[id].output_port_shape[slot]; in MaxMemoryShape()
223 gtl::InlinedVector<TensorShapeProto, 2> output_port_shape; member