Searched refs:UpSamplingNearestExp (Results 1 – 2 of 2) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mshadow/mshadow/extension/ |
| H A D | spatial_upsampling_nearest.h | 39 struct UpSamplingNearestExp : struct
40 public MakeTensorExp<UpSamplingNearestExp<SrcExp, DType, srcdim>,
47 UpSamplingNearestExp(const SrcExp &src, index_t scale) in UpSamplingNearestExp() argument
57 inline UpSamplingNearestExp<SrcExp, DType, ExpInfo<SrcExp>::kDim>
61 return UpSamplingNearestExp<SrcExp, DType, ExpInfo<SrcExp>::kDim>(src.self(), scale); in upsampling_nearest()
65 struct Plan<UpSamplingNearestExp<SrcExp, DType, srcdim>, DType> {
67 explicit Plan(const UpSamplingNearestExp<SrcExp, DType, srcdim> &e)
|
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/3rdparty/mshadow/mshadow/extension/ |
| H A D | spatial_upsampling_nearest.h | 39 struct UpSamplingNearestExp : struct
40 public MakeTensorExp<UpSamplingNearestExp<SrcExp, DType, srcdim>,
47 UpSamplingNearestExp(const SrcExp &src, index_t scale) in UpSamplingNearestExp() argument
57 inline UpSamplingNearestExp<SrcExp, DType, ExpInfo<SrcExp>::kDim>
61 return UpSamplingNearestExp<SrcExp, DType, ExpInfo<SrcExp>::kDim>(src.self(), scale); in upsampling_nearest()
65 struct Plan<UpSamplingNearestExp<SrcExp, DType, srcdim>, DType> {
67 explicit Plan(const UpSamplingNearestExp<SrcExp, DType, srcdim> &e)
|