Glow: is there 3D support?

Created on 22 Jan 2019  路  5Comments  路  Source: pytorch/glow

I see the data format is NCHW, it seems that this will only work with 2D image?
Is there support of 3D, such as NCDHW?

picture jingpengw

Most helpful comment

I can take a look at adding support for 3D

picture jackm321 on 23 Jan 2019
👍3

All 5 comments

Hi @jingpengw, no, currently there isn't support. 3D convolutions are of interest to us though. Would you be interested in working on this issue?

jfix71 picture jfix71 on 22 Jan 2019

yep, would be nice to have it, but this is not my priority right now.

jingpengw picture jingpengw on 22 Jan 2019

I can take a look at adding support for 3D

jackm321 picture jackm321 on 23 Jan 2019
👍3

@jackm321 what's the status of this?

nickgg picture nickgg on 30 Apr 2019

@jingpengw we added 3d convolution support https://github.com/pytorch/glow/blob/master/tools/ClassGen/NodeGen.cpp#L90
https://github.com/pytorch/glow/blob/master/lib/Backends/Interpreter/InterpreterNodes.cpp#L534

Can you flesh out more details about what you are looking for?

rdzhabarov picture rdzhabarov on 14 May 2019
Was this page helpful?
0 / 5 - 0 ratings

Related issues

dati91 picture dati91  路  3Comments
stoklund picture stoklund  路  5Comments
mciprian13 picture mciprian13  路  3Comments
rdzhabarov picture rdzhabarov  路  4Comments
tkclimb picture tkclimb  路  4Comments