Searched defs:PySGD (Results 1 – 1 of 1) sorted by relevance
30 struct PySGD final { struct31 explicit PySGD( in PySGD() function75 void step(const py::dict& py_dict) { in step()114 std::unique_ptr<optimizer::SGD> sgd_ = nullptr;115 std::vector<std::string> fqns_;118 std::vector<TensorPtr> params_;