Searched defs:is_cuda (Results 1 – 5 of 5) sorted by relevance
34 def _test_add(self, is_cuda=False, is_double=False): argument
59 void CodeGen_PyTorch::compile(const LoweredFunc &f, bool is_cuda) { in compile()
310 def _prediction_output(shape, dims, predts, is_cuda): argument
1429 bool is_cuda; member