| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/contrib/ |
| H A D | optimizer_op-inl.h | 99 DType *weight_data, const IType *grad_idx, const DType *grad_data, in Map()
105 auto get_data_j = [&i, &grad_idx, &row_length](index_t j) -> index_t { in Map()
106 return grad_idx[i] * row_length + j; in Map()
125 state_data[grad_idx[i]] += grad_ssq / row_length; in Map()
132 const DType div = lr * grad_rescaled / square_root::Map(state_data[grad_idx[i]] + eps); in Map()
166 const IType *grad_idx = grad.aux_data(rowsparse::kIdx).dptr<IType>(); in GroupAdagradUpdateDnsRspDnsImpl()
178 s, num_grad, row_length, out_data, state_data, weight_data, grad_idx, in GroupAdagradUpdateDnsRspDnsImpl()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/contrib/ |
| H A D | optimizer_op-inl.h | 99 DType *weight_data, const IType *grad_idx, const DType *grad_data, in Map()
105 auto get_data_j = [&i, &grad_idx, &row_length](index_t j) -> index_t { in Map()
106 return grad_idx[i] * row_length + j; in Map()
125 state_data[grad_idx[i]] += grad_ssq / row_length; in Map()
132 const DType div = lr * grad_rescaled / square_root::Map(state_data[grad_idx[i]] + eps); in Map()
166 const IType *grad_idx = grad.aux_data(rowsparse::kIdx).dptr<IType>(); in GroupAdagradUpdateDnsRspDnsImpl()
178 s, num_grad, row_length, out_data, state_data, weight_data, grad_idx, in GroupAdagradUpdateDnsRspDnsImpl()
|
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/ |
| H A D | optimizer_op.cu | 35 DType* mom_data, const DType* weight_data, const IType* grad_idx, in Map()
81 IType* grad_idx = grad.aux_data(kIdx).dptr<IType>(); in SGDMomStdUpdateDnsRspDnsImpl()
106 prefix_sum, grad_idx); in SGDMomStdUpdateDnsRspDnsImpl()
117 out_data, mom_data, weight_data, grad_idx, grad_val, prefix_sum, in SGDMomStdUpdateDnsRspDnsImpl()
130 DType* mean_data, DType* var_data, const DType* weight_data, const IType* grad_idx, in Map()
178 const IType* grad_idx = grad.aux_data(kIdx).dptr<IType>(); in AdamStdUpdateDnsRspDnsImpl()
203 prefix_sum, grad_idx); in AdamStdUpdateDnsRspDnsImpl()
214 row_length, out_data, mean_data, var_data, weight_data, grad_idx, grad_val, prefix_sum, in AdamStdUpdateDnsRspDnsImpl()
|
| H A D | optimizer_op.cc | 111 DType* mom_data, const DType* weight_data, const IType* grad_idx, in Map()
164 const IType* grad_idx = grad.aux_data(kIdx).dptr<IType>(); in SGDMomStdUpdateDnsRspDnsImpl()
178 prefix_sum, grad_idx); in SGDMomStdUpdateDnsRspDnsImpl()
185 out_data, mom_data, weight_data, grad_idx, grad_val, prefix_sum, in SGDMomStdUpdateDnsRspDnsImpl()
198 DType* mean_data, DType* var_data, const DType* weight_data, const IType* grad_idx, in Map()
254 const IType* grad_idx = grad.aux_data(kIdx).dptr<IType>(); in AdamStdUpdateDnsRspDnsImpl()
269 prefix_sum, grad_idx); in AdamStdUpdateDnsRspDnsImpl()
277 out_data, mean_data, var_data, weight_data, grad_idx, grad_val, prefix_sum, in AdamStdUpdateDnsRspDnsImpl()
|
| H A D | optimizer_op-inl.h | 439 const dim_t row_offset = grad_idx[row_id] * row_length;
464 index_t data_i = grad_idx[i] * row_length + j;
520 out->dptr<DType>(), weight_data, grad_idx, grad_val,
758 index_t data_i = grad_idx[i] * row_length + j;
826 IType* grad_idx = grad.aux_data(kIdx).dptr<IType>();
837 out_data, mom_data, weight_data, grad_idx, grad_val,
1359 const dim_t row_offset = grad_idx[i] * row_length;
1394 const dim_t row_offset = grad_idx[row_id] * row_length;
2194 const dim_t row_offset = grad_idx[i] * row_length;
2517 const dim_t data_i = grad_idx[i] * row_length;
[all …]
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/ |
| H A D | optimizer_op.cu | 35 DType* mom_data, const DType* weight_data, const IType* grad_idx, in Map()
81 IType* grad_idx = grad.aux_data(kIdx).dptr<IType>(); in SGDMomStdUpdateDnsRspDnsImpl()
106 prefix_sum, grad_idx); in SGDMomStdUpdateDnsRspDnsImpl()
117 out_data, mom_data, weight_data, grad_idx, grad_val, prefix_sum, in SGDMomStdUpdateDnsRspDnsImpl()
130 DType* mean_data, DType* var_data, const DType* weight_data, const IType* grad_idx, in Map()
178 const IType* grad_idx = grad.aux_data(kIdx).dptr<IType>(); in AdamStdUpdateDnsRspDnsImpl()
203 prefix_sum, grad_idx); in AdamStdUpdateDnsRspDnsImpl()
214 row_length, out_data, mean_data, var_data, weight_data, grad_idx, grad_val, prefix_sum, in AdamStdUpdateDnsRspDnsImpl()
|
| H A D | optimizer_op.cc | 111 DType* mom_data, const DType* weight_data, const IType* grad_idx, in Map()
164 const IType* grad_idx = grad.aux_data(kIdx).dptr<IType>(); in SGDMomStdUpdateDnsRspDnsImpl()
178 prefix_sum, grad_idx); in SGDMomStdUpdateDnsRspDnsImpl()
185 out_data, mom_data, weight_data, grad_idx, grad_val, prefix_sum, in SGDMomStdUpdateDnsRspDnsImpl()
198 DType* mean_data, DType* var_data, const DType* weight_data, const IType* grad_idx, in Map()
254 const IType* grad_idx = grad.aux_data(kIdx).dptr<IType>(); in AdamStdUpdateDnsRspDnsImpl()
269 prefix_sum, grad_idx); in AdamStdUpdateDnsRspDnsImpl()
277 out_data, mean_data, var_data, weight_data, grad_idx, grad_val, prefix_sum, in AdamStdUpdateDnsRspDnsImpl()
|
| H A D | optimizer_op-inl.h | 439 const dim_t row_offset = grad_idx[row_id] * row_length;
464 index_t data_i = grad_idx[i] * row_length + j;
520 out->dptr<DType>(), weight_data, grad_idx, grad_val,
758 index_t data_i = grad_idx[i] * row_length + j;
826 IType* grad_idx = grad.aux_data(kIdx).dptr<IType>();
837 out_data, mom_data, weight_data, grad_idx, grad_val,
1359 const dim_t row_offset = grad_idx[i] * row_length;
1394 const dim_t row_offset = grad_idx[row_id] * row_length;
2194 const dim_t row_offset = grad_idx[i] * row_length;
2517 const dim_t data_i = grad_idx[i] * row_length;
[all …]
|
| /dports/security/ridl/ridl-11b3240/stb/ |
| H A D | stb_perlin.h | 189 static float stb__perlin_grad(int grad_idx, float x, float y, float z) in stb__perlin_grad() argument
207 float *grad = basis[grad_idx]; in stb__perlin_grad()
|
| /dports/devel/stb/stb-f54acd4/ |
| H A D | stb_perlin.h | 191 static float stb__perlin_grad(int grad_idx, float x, float y, float z) in stb__perlin_grad() argument
209 float *grad = basis[grad_idx]; in stb__perlin_grad()
|
| /dports/graphics/aloadimage/arcan-0.6.1/src/engine/external/ |
| H A D | stb_perlin.h | 191 static float stb__perlin_grad(int grad_idx, float x, float y, float z) in stb__perlin_grad() argument
209 float *grad = basis[grad_idx]; in stb__perlin_grad()
|
| /dports/sysutils/acfgfs/arcan-0.6.1/src/engine/external/ |
| H A D | stb_perlin.h | 191 static float stb__perlin_grad(int grad_idx, float x, float y, float z) in stb__perlin_grad() argument
209 float *grad = basis[grad_idx]; in stb__perlin_grad()
|
| /dports/cad/cura-engine/CuraEngine-fadb5d6b/include/stb/ |
| H A D | stb_perlin.h | 190 static float stb__perlin_grad(int grad_idx, float x, float y, float z) in stb__perlin_grad() argument
208 float *grad = basis[grad_idx]; in stb__perlin_grad()
|
| /dports/devel/raylib/raylib-3.7.0/src/external/ |
| H A D | stb_perlin.h | 191 static float stb__perlin_grad(int grad_idx, float x, float y, float z) in stb__perlin_grad() argument
209 float *grad = basis[grad_idx]; in stb__perlin_grad()
|
| /dports/x11/aclip/arcan-0.6.1/src/engine/external/ |
| H A D | stb_perlin.h | 191 static float stb__perlin_grad(int grad_idx, float x, float y, float z) in stb__perlin_grad() argument
209 float *grad = basis[grad_idx]; in stb__perlin_grad()
|
| /dports/x11/arcan-trayicon/arcan-0.6.1/src/engine/external/ |
| H A D | stb_perlin.h | 191 static float stb__perlin_grad(int grad_idx, float x, float y, float z) in stb__perlin_grad() argument
209 float *grad = basis[grad_idx]; in stb__perlin_grad()
|
| /dports/science/py-dipy/dipy-1.4.1/dipy/tracking/ |
| H A D | life.py | 172 for grad_idx, this_grad in enumerate(grad):
173 tensors[grad_idx] = grad_tensor(this_grad, evals)
|