datasets_test.py | 135 def _testTransformation(self, fn): 150 dataset = fn(dataset, test_case['tensor'], test_case['shape']) 163 def fn(dataset, tensor, shape): function in function:GrapplerTest.testConcatenate 167 self._testTransformation(fn) 171 def fn(dataset, tensor, shape): function in function:GrapplerTest.testPrefetch 175 self._testTransformation(fn) 179 def fn(dataset, tensor, shape): function in function:GrapplerTest.testRepeat 183 self._testTransformation(fn) 187 def fn(dataset, tensor, shape): function in function:GrapplerTest.testShuffle 191 self._testTransformation(fn) 195 def fn(dataset, tensor, shape): function in function:GrapplerTest.testCache 203 def fn(dataset, tensor, shape): function in function:GrapplerTest.testTake 211 def fn(dataset, tensor, shape): function in function:GrapplerTest.testSkip 219 def fn(dataset, tensor, shape): function in function:GrapplerTest.testShard 227 def fn(dataset, tensor, shape): function in function:GrapplerTest.testFilter [all...] |