Searched defs:is_tpu (Results 1 – 2 of 2) sorted by relevance
43 def testTrainNetwork(self, distribution, optimizer_fn, is_tpu): argument
372 def testRunStepsWithOutputContext(self, distribution, optimizer_fn, is_tpu): argument