Searched refs:_tensorArrayWriteInWhile (Results 1 – 1 of 1) sorted by relevance
1839 def _tensorArrayWriteInWhile(self): member in TensorArrayBenchmark1850 op = self._tensorArrayWriteInWhile()1862 ds = ds.map(lambda _: self._tensorArrayWriteInWhile())1868 ds = ds.map(lambda _: self._tensorArrayWriteInWhile(), num_parallel_calls=2)