Searched defs:companion_obj (Results 1 – 1 of 1) sorted by relevance
243 def sharded_tensor_func(value, pg, device, companion_obj): argument260 def dtensor_func(value, pg, device, companion_obj): argument