Searched defs:GroupNormOptions (Results 1 – 2 of 2) sorted by relevance
149 struct TORCH_API GroupNormOptions { struct
11 GroupNormOptions::GroupNormOptions(int64_t num_groups, int64_t num_channels) in GroupNormOptions() function in torch::nn::GroupNormOptions