D | input_lib_test.py | 164 api_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 612 def testUnevenDatasetBatchesMultiWorker(self, input_type, api_type, argument 672 api_type, iteration_type, argument [all …]
|