| D | input_lib_test.py | 126 api_type, argument
280 def testOneDeviceCPU(self, input_type, api_type, iteration_type, distribution, argument
314 def testTwoDevicesOneGPUOneCPU(self, input_type, api_type, iteration_type, argument
345 def testTPU(self, input_type, api_type, iteration_type, distribution, argument
384 def testTupleDataset(self, input_type, api_type, iteration_type, distribution, argument
444 def testUnevenDatasetBatches(self, input_type, api_type, iteration_type, argument
483 def testBatchSplitting(self, input_type, api_type, iteration_type, argument
651 def testAutoshardingOption(self, input_type, api_type, iteration_type, argument
685 def testOneDevicePerWorker(self, input_type, api_type, iteration_type, argument
726 def testTwoDevicesPerWorker(self, input_type, api_type, iteration_type, argument
[all …]
|