projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: b23e4f6) | patch
[caffe2][easy] Save heap allocation in ConcatOp (#63529)
authorScott Wolchok <swolchok@fb.com>
Wed, 1 Sep 2021 01:22:23 +0000 (18:22 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Wed, 1 Sep 2021 01:33:32 +0000 (18:33 -0700)
commitc59970db6b7831c34053ecd5f86ce688bf68df7d
treef498f02ac39c450de6b055f247087bdc80283e51tree | snapshot
parentb23e4f6086f3174ae66243d77c032aef876c0246commit | diff
[caffe2][easy] Save heap allocation in ConcatOp (#63529)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/63529

Output() takes an IntArrayRef, so we can just use a std::initializer_list (stack-allocated array) instead of std::vector here.
ghstack-source-id: 137085908

Test Plan: existing CI

Reviewed By: mruberry

Differential Revision: D29687400

fbshipit-source-id: 9f2a7c6679f2552c098bb1bf7befaca18e0e5d4d
caffe2/operators/concat_split_op.h diff | blob | history
Domain: Machine Learning / ML Framework;