Searched defs:output_fn (Results 1 – 3 of 3) sorted by relevance
191 def output_fn(mod, outputs, device_mesh): function236 def output_fn(outputs, device_mesh): function
45 def output_fn(mod, outputs, device_mesh): function
118 def testRandomSeed(self, input_fn, output_fn): argument