Crate caffe2op_orderswitch

source ·
Expand description

| Note(Yangqing): I think it is possible to do | a more general swapaxes operator but I am | a little afraid of going down that general | path. Only implementing the two actually needed | ones here.

Structs

  • CudnnNCHW2NHWCOp
  • CudnnNHWC2NCHWOp
  • CudnnOrderSwithOpBase
  • GetNCHW2NHWCGradient
  • GetNHWC2NCHWGradient
  • NCHW2NHWCOp
    | The operator switches the order of data | in a tensor from NCHW- sample index N, | channels C, height H and width W, to the | NHWC order (this is for 2D images). | | In general, this operator switches | the order of data in a tensor from N C H_1 | … H_k to N H_1 … H_k C for k-dimensional | features, and currently supports k=1, | 2, and 3. |
  • NHWC2NCHWOp
    | The operator switches the order of data | in a tensor from NHWC- sample index N, | height H, width H and channels C, to the | NCHW order (this is for 2D images). | | In general, this operator switches | the order of data in a tensor from N H_1 | … H_k C to N C H_1 … H_k for k-dimensional | features, and currently supports k=1, | 2, and 3. |