Searched defs:on_gpu (Results 1 – 3 of 3) sorted by relevance
66 Scope on_gpu = root.WithAssignedDevice(kGPU).WithDevice(kGPU); in TEST() local93 Scope on_gpu = root.WithAssignedDevice(kGPU).WithDevice(kGPU); in TEST() local119 Scope on_gpu = root.WithAssignedDevice(kGPU).WithDevice(kGPU); in TEST() local151 Scope on_gpu = root.WithAssignedDevice(kGPU).WithDevice(kGPU); in TEST() local180 Scope on_gpu = root.WithAssignedDevice(kGPU).WithDevice(kGPU); in TEST() local
40 void ConstantOpTest::PersistentMemoryTrackingTest(bool on_gpu) { in PersistentMemoryTrackingTest()
3004 def on_gpu(x): function