Searched defs:tensor_len (Results 1 – 3 of 3) sorted by relevance
263 int num_devices, int num_subdivs, int tensor_len, in RunTest()
285 int num_devices, int num_subdivs, int tensor_len, in RunTest()
408 int num_devices, int tensor_len, int fail_after, in RunTest()