D | input_lib_test.py | 165 iteration_type, argument 366 def testOneDeviceCPU(self, input_type, api_type, iteration_type, distribution, argument 389 def testOneDeviceCPUMultiWorker(self, input_type, api_type, iteration_type, argument 420 def testTwoDevicesOneGPUOneCPU(self, input_type, api_type, iteration_type, argument 449 def testTPU(self, input_type, api_type, iteration_type, distribution, argument 485 def testTupleDataset(self, input_type, api_type, iteration_type, distribution, argument 520 def testTupleDatasetMultiworker(self, input_type, api_type, iteration_type, argument 577 def testUnevenDatasetBatches(self, input_type, api_type, iteration_type, argument 613 iteration_type, drop_remainder, argument 672 api_type, iteration_type, argument [all …]
|