Lines Matching full:is_cuda
77 def __call__(self, inputs, is_cuda, expect_fastpath, **kwargs): argument
81 is_cuda
142 def is_cuda(self): member in TestForeach
175 is_cuda=self.is_cuda,
184 is_cuda=self.is_cuda,
229 self.is_cuda,
261 actual = op(inputs, self.is_cuda, is_fastpath)
284 actual = op(inputs, self.is_cuda, is_fastpath, **op_kwargs)
344 [rhs_arg, tensors], is_cuda=False, expect_fastpath=False
401 if self.is_cuda:
506 actual = op(inputs, self.is_cuda, is_fastpath, **kwargs)
517 actual = op(inputs, self.is_cuda, is_fastpath, **kwargs)
561 (left_inputs, right_inputs), is_cuda=True, expect_fastpath=True
998 inputs, is_cuda=True, expect_fastpath=True, ord=ord, zero_size=False
1037 is_cuda=True,
1076 inputs, self.is_cuda, not disable_fastpath, zero_size=False, **kwargs
1143 is_cuda=False,
1168 is_cuda=False,
1321 (self_tensors, src_tensors), is_cuda=True, expect_fastpath=True