|
@@ -305,8 +305,8 @@ class InferImpl : public Infer {
|
|
|
input_buffer_.gpu(batch_size * input_numel);
|
|
input_buffer_.gpu(batch_size * input_numel);
|
|
|
output_array_.gpu(batch_size * num_classes_);
|
|
output_array_.gpu(batch_size * num_classes_);
|
|
|
output_array_.cpu(batch_size * num_classes_);
|
|
output_array_.cpu(batch_size * num_classes_);
|
|
|
- classes_indices_.gpu(batch_size)
|
|
|
|
|
- classes_indices_.cpu(batch_size)
|
|
|
|
|
|
|
+ classes_indices_.gpu(batch_size);
|
|
|
|
|
+ classes_indices_.cpu(batch_size);
|
|
|
|
|
|
|
|
|
|
|
|
|
if ((int)preprocess_buffers_.size() < batch_size) {
|
|
if ((int)preprocess_buffers_.size() < batch_size) {
|
