Searched defs:use_mps (Results 1 – 2 of 2) sorted by relevance
55 WORD32 use_mps; member
586 bool use_mps(const at::Tensor& input, const at::Tensor& weight) const { in use_mps() function