/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/ |
H A D | mshadow_op.h | 89 MSHADOW_XINLINE static DType Map(DType a, DType b) { \ 97 MSHADOW_XINLINE static DType Map(DType a, DType b) { \ 106 MSHADOW_XINLINE static DType Map(DType a, DType b) { \ 140 MSHADOW_XINLINE static DType Map(DType a, DType b) { in Map() 172 MSHADOW_XINLINE static DType Map(DType a, DType b) { in Map() 550 if (a < DType(0)) return DType(-DType(1)); in Map() 767 Map(DType a, DType b) { in Map() 790 Map(DType a, DType b) { in Map() 947 Map(DType a, DType b) { in Map() 970 Map(DType a, DType b) { in Map() [all …]
|
H A D | linalg.h | 59 void linalg_gemm(const Tensor<xpu, 2, DType>& A, const Tensor<xpu, 2, DType>& B, 60 const Tensor<xpu, 2, DType>& C, DType alpha, DType beta, 64 void linalg_batch_gemm(const Tensor<xpu, 3, DType>& A, const Tensor<xpu, 3, DType>& B, 65 const Tensor<xpu, 3, DType>& C, DType alpha, DType beta, 70 void linalg_batch_gemm(const Tensor<xpu, 4, DType>& A, const Tensor<xpu, 4, DType>& B, 71 const Tensor<xpu, 4, DType>& C, DType alpha, DType beta, 89 void linalg_trsm(const Tensor<xpu, 2, DType>& A, const Tensor<xpu, 2, DType>& B, 103 void linalg_trmm(const Tensor<xpu, 2, DType>& A, const Tensor<xpu, 2, DType>& B, 141 void linalg_syrk(const Tensor<xpu, 2, DType>& A, const Tensor<xpu, 2, DType>& B, 142 DType alpha, DType beta, bool tA, Stream<xpu> *s = 0); [all …]
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/ |
H A D | mshadow_op.h | 89 MSHADOW_XINLINE static DType Map(DType a, DType b) { \ 97 MSHADOW_XINLINE static DType Map(DType a, DType b) { \ 106 MSHADOW_XINLINE static DType Map(DType a, DType b) { \ 140 MSHADOW_XINLINE static DType Map(DType a, DType b) { in Map() 172 MSHADOW_XINLINE static DType Map(DType a, DType b) { in Map() 550 if (a < DType(0)) return DType(-DType(1)); in Map() 767 Map(DType a, DType b) { in Map() 790 Map(DType a, DType b) { in Map() 947 Map(DType a, DType b) { in Map() 970 Map(DType a, DType b) { in Map() [all …]
|
H A D | linalg.h | 59 void linalg_gemm(const Tensor<xpu, 2, DType>& A, const Tensor<xpu, 2, DType>& B, 60 const Tensor<xpu, 2, DType>& C, DType alpha, DType beta, 64 void linalg_batch_gemm(const Tensor<xpu, 3, DType>& A, const Tensor<xpu, 3, DType>& B, 65 const Tensor<xpu, 3, DType>& C, DType alpha, DType beta, 70 void linalg_batch_gemm(const Tensor<xpu, 4, DType>& A, const Tensor<xpu, 4, DType>& B, 71 const Tensor<xpu, 4, DType>& C, DType alpha, DType beta, 89 void linalg_trsm(const Tensor<xpu, 2, DType>& A, const Tensor<xpu, 2, DType>& B, 103 void linalg_trmm(const Tensor<xpu, 2, DType>& A, const Tensor<xpu, 2, DType>& B, 141 void linalg_syrk(const Tensor<xpu, 2, DType>& A, const Tensor<xpu, 2, DType>& B, 142 DType alpha, DType beta, bool tA, Stream<xpu> *s = 0); [all …]
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/nn/ |
H A D | pool_utils.h | 41 static MSHADOW_XINLINE DType Map(const DType a) { 48 static MSHADOW_XINLINE DType Map(const DType a) { 55 static MSHADOW_XINLINE DType Map(const DType a) { 62 static MSHADOW_XINLINE DType Map(const DType a) { 69 static MSHADOW_XINLINE DType Map(const DType a) { 76 static MSHADOW_XINLINE DType Map(const DType a) { 83 static MSHADOW_XINLINE DType Map(const DType a) { 97 static MSHADOW_XINLINE DType Map(const DType grad, const DType in_data, const DType out_data) { 104 static MSHADOW_XINLINE DType Map(const DType grad, const DType in_data, const DType out_data) { 111 static MSHADOW_XINLINE DType Map(const DType grad, const DType in_data, const DType out_data) { [all …]
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/nn/ |
H A D | pool_utils.h | 41 static MSHADOW_XINLINE DType Map(const DType a) { 48 static MSHADOW_XINLINE DType Map(const DType a) { 55 static MSHADOW_XINLINE DType Map(const DType a) { 62 static MSHADOW_XINLINE DType Map(const DType a) { 69 static MSHADOW_XINLINE DType Map(const DType a) { 76 static MSHADOW_XINLINE DType Map(const DType a) { 83 static MSHADOW_XINLINE DType Map(const DType a) { 97 static MSHADOW_XINLINE DType Map(const DType grad, const DType in_data, const DType out_data) { 104 static MSHADOW_XINLINE DType Map(const DType grad, const DType in_data, const DType out_data) { 111 static MSHADOW_XINLINE DType Map(const DType grad, const DType in_data, const DType out_data) { [all …]
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/contrib/ |
H A D | deformable_psroi_pooling.cc | 26 inline DType bilinear_interp_cpu(const DType* data, in bilinear_interp_cpu() 27 const DType x, const DType y, in bilinear_interp_cpu() 33 DType dist_x = static_cast<DType>(x - x1); in bilinear_interp_cpu() 34 DType dist_y = static_cast<DType>(y - y1); in bilinear_interp_cpu() 79 DType bin_size_w = roi_width / static_cast<DType>(pooled_width); in DeformablePSROIPoolForwardCPU() 87 DType trans_x = no_trans ? static_cast<DType>(0) : in DeformablePSROIPoolForwardCPU() 91 DType trans_y = no_trans ? static_cast<DType>(0) : in DeformablePSROIPoolForwardCPU() 96 DType wstart = static_cast<DType>(pw) * bin_size_w + roi_start_w; in DeformablePSROIPoolForwardCPU() 101 DType sum = 0; in DeformablePSROIPoolForwardCPU() 214 DType trans_x = no_trans ? static_cast<DType>(0) : in DeformablePSROIPoolBackwardAccCPU() [all …]
|
H A D | deformable_psroi_pooling.cu | 28 __device__ DType bilinear_interp(const DType* data, in bilinear_interp() 29 const DType x, const DType y, in bilinear_interp() 35 DType dist_x = static_cast<DType>(x - x1); in bilinear_interp() 36 DType dist_y = static_cast<DType>(y - y1); in bilinear_interp() 85 DType bin_size_w = roi_width / static_cast<DType>(pooled_width); in DeformablePSROIPoolForwardKernel() 93 DType trans_x = no_trans ? static_cast<DType>(0) : in DeformablePSROIPoolForwardKernel() 97 DType trans_y = no_trans ? static_cast<DType>(0) : in DeformablePSROIPoolForwardKernel() 102 DType wstart = static_cast<DType>(pw) * bin_size_w + roi_start_w; in DeformablePSROIPoolForwardKernel() 107 DType sum = 0; in DeformablePSROIPoolForwardKernel() 226 DType trans_x = no_trans ? static_cast<DType>(0) : in DeformablePSROIPoolBackwardAccKernel() [all …]
|
H A D | psroi_pooling.cu | 29 template <typename DType> 32 const DType* bottom_data, in PSROIPoolForwardKernel() 40 DType* top_data) { in PSROIPoolForwardKernel() 61 DType bin_size_h = roi_height / static_cast<DType>(pooled_height); in PSROIPoolForwardKernel() 62 DType bin_size_w = roi_width / static_cast<DType>(pooled_width); in PSROIPoolForwardKernel() 86 DType out_sum = 0; in PSROIPoolForwardKernel() 99 template<typename DType> 135 DType* bottom_diff, in PSROIPoolBackwardAccKernel() 157 DType bin_size_h = roi_height / static_cast<DType>(pooled_height); in PSROIPoolBackwardAccKernel() 158 DType bin_size_w = roi_width / static_cast<DType>(pooled_width); in PSROIPoolBackwardAccKernel() [all …]
|
H A D | psroi_pooling.cc | 25 template <typename DType> 28 const DType* bottom_data, in PSROIPoolForwardCPU() 36 DType* top_data) { in PSROIPoolForwardCPU() 56 DType roi_height = max(roi_end_h - roi_start_h, static_cast<DType>(0.1)); in PSROIPoolForwardCPU() 59 DType bin_size_h = roi_height / static_cast<DType>(pooled_height); in PSROIPoolForwardCPU() 60 DType bin_size_w = roi_width / static_cast<DType>(pooled_width); in PSROIPoolForwardCPU() 84 DType out_sum = 0; in PSROIPoolForwardCPU() 131 DType* bottom_diff, in PSROIPoolBackwardAccCPU() 153 DType bin_size_h = roi_height / static_cast<DType>(pooled_height); in PSROIPoolBackwardAccCPU() 154 DType bin_size_w = roi_width / static_cast<DType>(pooled_width); in PSROIPoolBackwardAccCPU() [all …]
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/contrib/ |
H A D | deformable_psroi_pooling.cc | 26 inline DType bilinear_interp_cpu(const DType* data, in bilinear_interp_cpu() 27 const DType x, const DType y, in bilinear_interp_cpu() 33 DType dist_x = static_cast<DType>(x - x1); in bilinear_interp_cpu() 34 DType dist_y = static_cast<DType>(y - y1); in bilinear_interp_cpu() 79 DType bin_size_w = roi_width / static_cast<DType>(pooled_width); in DeformablePSROIPoolForwardCPU() 87 DType trans_x = no_trans ? static_cast<DType>(0) : in DeformablePSROIPoolForwardCPU() 91 DType trans_y = no_trans ? static_cast<DType>(0) : in DeformablePSROIPoolForwardCPU() 96 DType wstart = static_cast<DType>(pw) * bin_size_w + roi_start_w; in DeformablePSROIPoolForwardCPU() 101 DType sum = 0; in DeformablePSROIPoolForwardCPU() 214 DType trans_x = no_trans ? static_cast<DType>(0) : in DeformablePSROIPoolBackwardAccCPU() [all …]
|
H A D | deformable_psroi_pooling.cu | 28 __device__ DType bilinear_interp(const DType* data, in bilinear_interp() 29 const DType x, const DType y, in bilinear_interp() 35 DType dist_x = static_cast<DType>(x - x1); in bilinear_interp() 36 DType dist_y = static_cast<DType>(y - y1); in bilinear_interp() 85 DType bin_size_w = roi_width / static_cast<DType>(pooled_width); in DeformablePSROIPoolForwardKernel() 93 DType trans_x = no_trans ? static_cast<DType>(0) : in DeformablePSROIPoolForwardKernel() 97 DType trans_y = no_trans ? static_cast<DType>(0) : in DeformablePSROIPoolForwardKernel() 102 DType wstart = static_cast<DType>(pw) * bin_size_w + roi_start_w; in DeformablePSROIPoolForwardKernel() 107 DType sum = 0; in DeformablePSROIPoolForwardKernel() 226 DType trans_x = no_trans ? static_cast<DType>(0) : in DeformablePSROIPoolBackwardAccKernel() [all …]
|
H A D | psroi_pooling.cu | 29 template <typename DType> 32 const DType* bottom_data, in PSROIPoolForwardKernel() 40 DType* top_data) { in PSROIPoolForwardKernel() 61 DType bin_size_h = roi_height / static_cast<DType>(pooled_height); in PSROIPoolForwardKernel() 62 DType bin_size_w = roi_width / static_cast<DType>(pooled_width); in PSROIPoolForwardKernel() 86 DType out_sum = 0; in PSROIPoolForwardKernel() 99 template<typename DType> 135 DType* bottom_diff, in PSROIPoolBackwardAccKernel() 157 DType bin_size_h = roi_height / static_cast<DType>(pooled_height); in PSROIPoolBackwardAccKernel() 158 DType bin_size_w = roi_width / static_cast<DType>(pooled_width); in PSROIPoolBackwardAccKernel() [all …]
|
H A D | psroi_pooling.cc | 25 template <typename DType> 28 const DType* bottom_data, in PSROIPoolForwardCPU() 36 DType* top_data) { in PSROIPoolForwardCPU() 56 DType roi_height = max(roi_end_h - roi_start_h, static_cast<DType>(0.1)); in PSROIPoolForwardCPU() 59 DType bin_size_h = roi_height / static_cast<DType>(pooled_height); in PSROIPoolForwardCPU() 60 DType bin_size_w = roi_width / static_cast<DType>(pooled_width); in PSROIPoolForwardCPU() 84 DType out_sum = 0; in PSROIPoolForwardCPU() 131 DType* bottom_diff, in PSROIPoolBackwardAccCPU() 153 DType bin_size_h = roi_height / static_cast<DType>(pooled_height); in PSROIPoolBackwardAccCPU() 154 DType bin_size_w = roi_width / static_cast<DType>(pooled_width); in PSROIPoolBackwardAccCPU() [all …]
|
/dports/devel/dbus-sharp/dbus-sharp-0.8.1/src/Protocol/ |
H A D | Signature.cs | 171 case DType.Byte: in DataForDType() 308 case DType.Byte: in GetSize() 312 case DType.Int16: in GetSize() 720 DType dtype = (DType)data[pos++]; in GetNextSignature() 727 if ((DType)data[pos] == DType.DictEntryBegin) { in GetNextSignature() 743 while ((DType)data[pos] != DType.StructEnd) in GetNextSignature() 766 DType dtype = (DType)data[pos++]; in ToType() 771 case DType.Byte: in ToType() 799 if ((DType)data[pos] == DType.DictEntryBegin) { in ToType() 812 while (((DType)data[pos]) != DType.StructEnd) in ToType() [all …]
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mshadow/mshadow/ |
H A D | base.h | 557 MSHADOW_XINLINE static DType Map(DType a, DType b) { 565 MSHADOW_XINLINE static DType Map(DType a, DType b) { 573 MSHADOW_XINLINE static DType Map(DType a, DType b) { 581 MSHADOW_XINLINE static DType Map(DType a, DType b) { 589 MSHADOW_XINLINE static DType Map(DType a, DType b) { 601 MSHADOW_XINLINE static DType Map(DType a) { 923 …INLINE static void Merge(volatile DType& dst_val, volatile DType& dst_residual, volatile DType& sr… 946 MSHADOW_XINLINE static DType PartialGrad(DType redres, DType redsrc) { 986 …INLINE static void Merge(volatile DType& dst_val, volatile DType& dst_residual, volatile DType& sr… 1000 MSHADOW_XINLINE static DType PartialGrad(DType redres, DType redsrc) { [all …]
|
H A D | expression.h | 95 struct ScalarExp: public Exp<ScalarExp<DType>, DType, type::kMapper> { 97 DType scalar_; 103 inline ScalarExp<DType> scalar(DType s) { in scalar() 231 DType scale_; 240 dot(const RValueExp<TA, DType> &lhs, const RValueExp<TB, DType> &rhs) { in dot() 246 dot(const TransposeExp<TA, DType> &lhs, const RValueExp<TB, DType> &rhs) { in dot() 252 dot(const RValueExp<TA, DType> &lhs, const TransposeExp<TB, DType> &rhs) { in dot() 295 MakeExp(const Exp<TA, DType, ta> &item1, const Exp<TB, DType, tb> &item2, in MakeExp() argument 319 F(const Exp<TA, DType, ta> &item1, const Exp<TB, DType, tb> &item2, in F() argument 348 MakeExp(const Exp<TA, DType, ta> &lhs, const Exp<TB, DType, tb> &rhs) { in MakeExp() argument [all …]
|
H A D | expr_engine-inl.h | 68 class Plan<Tensor<Device, dim, DType>, DType> { 87 class Plan<Tensor<Device, 1, DType>, DType> { 102 class Plan<ScalarExp<DType>, DType> { 144 class Plan<BinaryMapExp<OP, TA, TB, DType, etype>, DType> { 158 class Plan<UnaryMapExp<OP, TA, DType, etype>, DType> { 182 class Plan<TransposeExp<EType, DType>, DType> { 196 inline Plan<BinaryMapExp<OP, TA, TB, DType, etype>, DType> 204 inline Plan<ScalarExp<DType>, DType> MakePlan(const ScalarExp<DType> &e) { 205 return Plan<ScalarExp<DType>, DType>(e.scalar_); 220 inline Plan<TransposeExp<T, DType>, DType> [all …]
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/3rdparty/mshadow/mshadow/ |
H A D | base.h | 557 MSHADOW_XINLINE static DType Map(DType a, DType b) { 565 MSHADOW_XINLINE static DType Map(DType a, DType b) { 573 MSHADOW_XINLINE static DType Map(DType a, DType b) { 581 MSHADOW_XINLINE static DType Map(DType a, DType b) { 589 MSHADOW_XINLINE static DType Map(DType a, DType b) { 601 MSHADOW_XINLINE static DType Map(DType a) { 923 …INLINE static void Merge(volatile DType& dst_val, volatile DType& dst_residual, volatile DType& sr… 946 MSHADOW_XINLINE static DType PartialGrad(DType redres, DType redsrc) { 986 …INLINE static void Merge(volatile DType& dst_val, volatile DType& dst_residual, volatile DType& sr… 1000 MSHADOW_XINLINE static DType PartialGrad(DType redres, DType redsrc) { [all …]
|
H A D | expression.h | 95 struct ScalarExp: public Exp<ScalarExp<DType>, DType, type::kMapper> { 97 DType scalar_; 103 inline ScalarExp<DType> scalar(DType s) { in scalar() 231 DType scale_; 240 dot(const RValueExp<TA, DType> &lhs, const RValueExp<TB, DType> &rhs) { in dot() 246 dot(const TransposeExp<TA, DType> &lhs, const RValueExp<TB, DType> &rhs) { in dot() 252 dot(const RValueExp<TA, DType> &lhs, const TransposeExp<TB, DType> &rhs) { in dot() 295 MakeExp(const Exp<TA, DType, ta> &item1, const Exp<TB, DType, tb> &item2, in MakeExp() argument 319 F(const Exp<TA, DType, ta> &item1, const Exp<TB, DType, tb> &item2, in F() argument 348 MakeExp(const Exp<TA, DType, ta> &lhs, const Exp<TB, DType, tb> &rhs) { in MakeExp() argument [all …]
|
H A D | packet-inl.h | 246 class PacketPlan<Tensor<Device, dim, DType>, DType, Arch> { 258 const DType *dptr_; 263 class PacketPlan<ScalarExp<DType>, DType, Arch> { 274 DType scalar_; 278 class PacketPlan<BinaryMapExp<OP, TA, TB, DType, etype>, DType, Arch> { 295 class PacketPlan<UnaryMapExp<OP, TA, DType, etype>, DType, Arch> { 310 inline PacketPlan<BinaryMapExp<OP, TA, TB, DType, etype>, DType, Arch> 314 inline PacketPlan<ScalarExp<DType>, DType, Arch> MakePacketPlan(const ScalarExp<DType> &e) { 315 return PacketPlan<ScalarExp<DType>, DType, Arch>(e.scalar_); 327 inline PacketPlan<UnaryMapExp<OP, TA, DType, etype>, DType, Arch> [all …]
|
H A D | expr_engine-inl.h | 68 class Plan<Tensor<Device, dim, DType>, DType> { 87 class Plan<Tensor<Device, 1, DType>, DType> { 102 class Plan<ScalarExp<DType>, DType> { 144 class Plan<BinaryMapExp<OP, TA, TB, DType, etype>, DType> { 158 class Plan<UnaryMapExp<OP, TA, DType, etype>, DType> { 182 class Plan<TransposeExp<EType, DType>, DType> { 196 inline Plan<BinaryMapExp<OP, TA, TB, DType, etype>, DType> 204 inline Plan<ScalarExp<DType>, DType> MakePlan(const ScalarExp<DType> &e) { 205 return Plan<ScalarExp<DType>, DType>(e.scalar_); 220 inline Plan<TransposeExp<T, DType>, DType> [all …]
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/tensor/ |
H A D | la_op-inl.h | 63 MSHADOW_XINLINE static void Map(int i, DType scale, DType* data) { in Map() 75 const Tensor<xpu, dim, DType>& C, DType alpha, DType beta, in op() 103 gemm::op(A, B, C, DType(alpha), DType(0), tA, tB, s); in op() 216 DType sum(0); in Map() 349 op(A, B, DType(param.alpha), DType(0), param.transpose, s); in op() 477 DType *LU, DType* sign, DType *logdet) { in Map() 486 diag_sign *= ((DType(0) < diag) - (diag < DType(0))); in Map() 646 gemm::op(A, dA, dB, DType(1.), DType(1.), false, true, s); in op() 729 MSHADOW_XINLINE static void Map(int i, int M, int stride, DType* dB, DType* A, DType* dA) { in Map() 817 DType syevd_back_helper_eps(DType* X); [all …]
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/tensor/ |
H A D | la_op-inl.h | 63 MSHADOW_XINLINE static void Map(int i, DType scale, DType* data) { in Map() 75 const Tensor<xpu, dim, DType>& C, DType alpha, DType beta, in op() 103 gemm::op(A, B, C, DType(alpha), DType(0), tA, tB, s); in op() 216 DType sum(0); in Map() 349 op(A, B, DType(param.alpha), DType(0), param.transpose, s); in op() 477 DType *LU, DType* sign, DType *logdet) { in Map() 486 diag_sign *= ((DType(0) < diag) - (diag < DType(0))); in Map() 646 gemm::op(A, dA, dB, DType(1.), DType(1.), false, true, s); in op() 729 MSHADOW_XINLINE static void Map(int i, int M, int stride, DType* dB, DType* A, DType* dA) { in Map() 817 DType syevd_back_helper_eps(DType* X); [all …]
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mshadow/mshadow/extension/ |
H A D | complex.h | 38 MSHADOW_XINLINE static DType RealMap(DType a_real, DType a_imag, in RealMap() 39 DType b_real, DType b_imag) { in RealMap() 43 MSHADOW_XINLINE static DType ImagMap(DType a_real, DType a_imag, in ImagMap() 44 DType b_real, DType b_imag) { in ImagMap() 52 MSHADOW_XINLINE static DType RealMap(DType a_real, DType a_imag, in RealMap() 53 DType b_real, DType b_imag) { in RealMap() 57 MSHADOW_XINLINE static DType ImagMap(DType a_real, DType a_imag, in ImagMap() 58 DType b_real, DType b_imag) { in ImagMap() 65 MSHADOW_XINLINE static DType RealMap(const expr::Plan<TA, DType> &src_, in RealMap() 149 DType, etype> { [all …]
|