Extend ONNX while op by x2, rather than x1.02
Summary:
I think the original author wrote 2.0f in attempt to double in size, but this argument takes a percentage increase, not a factor increase.
Created from Diffusion's 'Open in Editor' feature.
Reviewed By: jamesr66a
Differential Revision: D10412946
fbshipit-source-id: 95eb3d284255f232b7782bb1d2c9c2ef8aa6f8a7
diff --git a/caffe2/operators/onnx_while_op.h b/caffe2/operators/onnx_while_op.h
index f55a4a0..4280399 100644
--- a/caffe2/operators/onnx_while_op.h
+++ b/caffe2/operators/onnx_while_op.h
@@ -171,7 +171,7 @@
scan_outputs_sizes[i],
"Size of scan output changed across iterations");
dims.insert(dims.begin(), itr);
- scan_output_target->Extend(1, 2.0f, &context_);
+ scan_output_target->Extend(1, 100, &context_);
int64_t timestep_size = 1;
for (const int64_t t : scan_outputs_sizes[i]) {
