File serialize.h

↰ Parent directory (torch/csrc/api/include/torch/optim)

Definition (torch/csrc/api/include/torch/optim/serialize.h)

• Program Listing for File serialize.h

Includes

• c10/util/irange.h

• cstddef

• cstdint

• deque

• string

• torch/optim/optimizer.h (File optimizer.h)

• torch/serialize/archive.h (File archive.h)

• torch/types.h (File types.h)

• vector

Included By

• File adagrad.h

• File adam.h

• File adamw.h

• File lbfgs.h

• File rmsprop.h

• File sgd.h

Namespaces

• Namespace torch

• Namespace torch::optim

• Namespace torch::optim::detail

Functions

• Template Function torch::optim::deque_to_list

• Template Function torch::optim::detail::serialize(serialize::InputArchive&, ska::flat_hash_map<void *, std::unique_ptr<OptimizerParamState>>&)

• Template Function torch::optim::detail::serialize(serialize::OutputArchive&, const std::vector<OptimizerParamGroup>&)

• Template Function torch::optim::detail::serialize(serialize::InputArchive&, std::vector<std::pair<std::vector<std::string>, std::unique_ptr<OptimizerOptions>>>&)

• Template Function torch::optim::detail::serialize(serialize::OutputArchive&, const ska::flat_hash_map<void *, std::unique_ptr<OptimizerParamState>>&)

• Template Function torch::optim::list_to_deque

• Function torch::optim::serialize(serialize::OutputArchive&, const std::string&, const int64_t&)

• Function torch::optim::serialize(serialize::InputArchive&, const std::string&, int64_t&)

• Function torch::optim::serialize(serialize::OutputArchive&, const std::string&, const std::vector<int64_t>&)

• Function torch::optim::serialize(serialize::InputArchive&, const std::string&, std::vector<int64_t>&)

• Template Function torch::optim::serialize(serialize::OutputArchive&, const Optimizer&)

• Template Function torch::optim::serialize(serialize::InputArchive&, Optimizer&)

• Template Function torch::optim::serialize(serialize::OutputArchive&, const std::string&, const BufferContainer&)

• Template Function torch::optim::serialize(serialize::InputArchive&, const std::string&, BufferContainer&)

Defines

• Define _TORCH_OPTIM_DESERIALIZE_TORCH_ARG

• Define _TORCH_OPTIM_DESERIALIZE_TORCH_ARG_DEQUE

• Define _TORCH_OPTIM_DESERIALIZE_TORCH_ARG_OPTIONAL

• Define _TORCH_OPTIM_SERIALIZE

• Define _TORCH_OPTIM_SERIALIZE_TORCH_ARG

• Define _TORCH_OPTIM_SERIALIZE_TORCH_ARG_DEQUE

• Define _TORCH_OPTIM_SERIALIZE_WITH_TEMPLATE_ARG