mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 05:34:18 +08:00
Renaming dims() to sizes() (fbcode)
Summary: Codemod generated with clangr shard mode, 25 files per diff, for renaming dims() to sizes() Reviewed By: ezyang Differential Revision: D10848643 fbshipit-source-id: ac75833be8be9162e35b00dcd352f616bc7bbafe
This commit is contained in:
committed by
Facebook Github Bot
parent
1720757220
commit
e5752f2cb4
@ -37,7 +37,7 @@ class UpsampleNearestOp final : public Operator<Context> {
|
||||
bool RunOnDevice() override {
|
||||
auto& X = Input(0);
|
||||
auto* Y = Output(0);
|
||||
auto out_shape = X.dims().vec();
|
||||
auto out_shape = X.sizes().vec();
|
||||
out_shape[X.ndim() - 1] *= scale_;
|
||||
out_shape[X.ndim() - 2] *= scale_;
|
||||
Y->Resize(out_shape);
|
||||
|
Reference in New Issue
Block a user