Searched refs:MultiSGDMomUpdate (Results 1 – 6 of 6) sorted by relevance
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/ |
H A D | optimizer_op.cu | 247 .set_attr<FCompute>("FCompute<gpu>", MultiSGDMomUpdate<gpu, type_identity, 3>); 251 .set_attr<FCompute>("FCompute<gpu>", MultiSGDMomUpdate<gpu, single_precision, 4>);
|
H A D | optimizer_op.cc | 405 .set_attr<FCompute>("FCompute<cpu>", MultiSGDMomUpdate<cpu, type_identity, 3>) 505 .set_attr<FCompute>("FCompute<cpu>", MultiSGDMomUpdate<cpu, single_precision, 4>)
|
H A D | optimizer_op-inl.h | 360 inline void MultiSGDMomUpdate(const nnvm::NodeAttrs& attrs, in MultiSGDMomUpdate() function
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/ |
H A D | optimizer_op.cu | 247 .set_attr<FCompute>("FCompute<gpu>", MultiSGDMomUpdate<gpu, type_identity, 3>); 251 .set_attr<FCompute>("FCompute<gpu>", MultiSGDMomUpdate<gpu, single_precision, 4>);
|
H A D | optimizer_op.cc | 405 .set_attr<FCompute>("FCompute<cpu>", MultiSGDMomUpdate<cpu, type_identity, 3>) 505 .set_attr<FCompute>("FCompute<cpu>", MultiSGDMomUpdate<cpu, single_precision, 4>)
|
H A D | optimizer_op-inl.h | 360 inline void MultiSGDMomUpdate(const nnvm::NodeAttrs& attrs, in MultiSGDMomUpdate() function
|