Home
last modified time | relevance | path

Searched refs:DType (Results 1 – 25 of 1628) sorted by relevance

12345678910>>...66

/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/
H A Dmshadow_op.h89 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 Dlinalg.h59 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 Dmshadow_op.h89 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 Dlinalg.h59 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 Dpool_utils.h41 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 Dpool_utils.h41 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 Ddeformable_psroi_pooling.cc26 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 Ddeformable_psroi_pooling.cu28 __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 Dpsroi_pooling.cu29 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 Dpsroi_pooling.cc25 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 Ddeformable_psroi_pooling.cc26 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 Ddeformable_psroi_pooling.cu28 __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 Dpsroi_pooling.cu29 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 Dpsroi_pooling.cc25 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 DSignature.cs171 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 Dbase.h557 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 Dexpression.h95 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 Dexpr_engine-inl.h68 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 Dbase.h557 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 Dexpression.h95 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 Dpacket-inl.h246 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 Dexpr_engine-inl.h68 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 Dla_op-inl.h63 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 Dla_op-inl.h63 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 Dcomplex.h38 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 …]

12345678910>>...66