Home
last modified time | relevance | path

Searched refs:record_fn (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dinput_ops_test.py103 def _verifySimpleShardingOutput(self, dataset, record_fn): argument
108 self.assertAllEqual(record_fn(r, f), self.evaluate(next_element_fn()))
209 record_fn = lambda r, f: (self._record(r, f), self._text_line(r, f)) function
210 self._verifySimpleShardingOutput(dataset, record_fn)
/external/tensorflow/tensorflow/python/kernel_tests/
Dsummary_ops_test.py431 def record_fn(): function
435 with summary_ops.record_if(record_fn):
453 def record_fn(): function
459 with summary_ops.record_if(record_fn):