Home
last modified time | relevance | path

Searched refs:MultiSGDMomUpdate (Results 1 – 6 of 6) sorted by relevance

/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/
H A Doptimizer_op.cu247 .set_attr<FCompute>("FCompute<gpu>", MultiSGDMomUpdate<gpu, type_identity, 3>);
251 .set_attr<FCompute>("FCompute<gpu>", MultiSGDMomUpdate<gpu, single_precision, 4>);
H A Doptimizer_op.cc405 .set_attr<FCompute>("FCompute<cpu>", MultiSGDMomUpdate<cpu, type_identity, 3>)
505 .set_attr<FCompute>("FCompute<cpu>", MultiSGDMomUpdate<cpu, single_precision, 4>)
H A Doptimizer_op-inl.h360 inline void MultiSGDMomUpdate(const nnvm::NodeAttrs& attrs, in MultiSGDMomUpdate() function
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/
H A Doptimizer_op.cu247 .set_attr<FCompute>("FCompute<gpu>", MultiSGDMomUpdate<gpu, type_identity, 3>);
251 .set_attr<FCompute>("FCompute<gpu>", MultiSGDMomUpdate<gpu, single_precision, 4>);
H A Doptimizer_op.cc405 .set_attr<FCompute>("FCompute<cpu>", MultiSGDMomUpdate<cpu, type_identity, 3>)
505 .set_attr<FCompute>("FCompute<cpu>", MultiSGDMomUpdate<cpu, single_precision, 4>)
H A Doptimizer_op-inl.h360 inline void MultiSGDMomUpdate(const nnvm::NodeAttrs& attrs, in MultiSGDMomUpdate() function