Home | History | Annotate | Download | only in framework

Lines Matching refs:gpu_index

804   def gpu(self, gpu_index=0):
808 gpu_index: Identifies which GPU to place the contents on the returned
815 return self._copy(context.context(), "GPU:" + str(gpu_index))