Searched refs:my_complex_tensor_function (Results 1 – 1 of 1) sorted by relevance
266 def my_complex_tensor_function(list_input, tensor_class_input, dict_input): function2602 worker_name(dst_rank), my_complex_tensor_function, args=(a, b, c)2604 self.assertEqual(ret, my_complex_tensor_function(a, b, c))