Searched defs:ReLU6Options (Results 1 – 2 of 2) sorted by relevance
397 struct TORCH_API ReLU6Options { struct
20 ReLU6Options::ReLU6Options(bool inplace) : inplace_(inplace) {} in ReLU6Options() function in torch::nn::ReLU6Options