Searched defs:assign_device (Results 1 – 2 of 2) sorted by relevance
49 def assign_device(cls, core): member in Sharding174 def assign_device(tensor, device): function
57 void SwapGraph(Graph* graph, bool assign_device = false) { in SwapGraph()