Searched defs:param_group (Results 1 – 14 of 14) sorted by relevance
44 SGDParamGroup(const SGDParamGroup& param_group) in SGDParamGroup()80 for (const auto& param_group : param_groups) { in SGD() local
49 void SGD::add_param_group(const SGDParamGroup& param_group) { in add_param_group()
84 OptimizerParamGroup(const OptimizerParamGroup& param_group) in OptimizerParamGroup()121 for (const auto& param_group : param_groups) { in Optimizer() local
135 SGDParamGroup(const SGDParamGroup& param_group) in SGDParamGroup()186 for (const auto& param_group : param_groups) { in SGD() local
74 void SGD::add_param_group(const SGDParamGroup& param_group) { in add_param_group()
109 def add_param_group(self, param_group): argument
44 param_group: str = "group" variable in OpConv2d153 param_group = "group" variable in OpGroupNorm379 param_group: str = "group" variable in OpTransposeConv2d
185 def get_hook(param_group, i): argument
98 void Optimizer::add_param_group(const OptimizerParamGroup& param_group) { in add_param_group()
682 const auto& param_group = groups->GetGroup(param_group_index->first); in AddCopiesToRoot() local
105 for (auto& param_group : optim3_2.param_groups()) { in test_serialize_optimizer() local
145 __u32 param_group; member
96350 struct config_group param_group; member