Searched refs:device_only_candidate (Results 1 – 1 of 1) sorted by relevance
4404 op, device_only_candidate = _op_to_colocate_with(op, self)4424 if device_only_candidate is not None:4425 self._colocation_stack.push_obj(device_only_candidate, offset=4)4437 if device_only_candidate is not None:6828 device_only_candidate = lambda: None function6829 device_only_candidate.device = v.device6830 device_only_candidate.name = v.name6832 return graph.capture(v.handle).op, device_only_candidate6834 return v.handle.op, device_only_candidate