Searched defs:LayerNormOptions (Results 1 – 2 of 2) sorted by relevance
18 struct TORCH_API LayerNormOptions { struct
6 LayerNormOptions::LayerNormOptions(std::vector<int64_t> normalized_shape) in LayerNormOptions() function in torch::nn::LayerNormOptions