Searched refs:dev0_tensors (Results 1 – 1 of 1) sorted by relevance
203 def loop_body(dev0_tensors, dev1_tensors, loop_tensor): argument207 device_tensors = dev0_tensors if i == 0 else dev1_tensors