Searched refs:profile_batch (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/ |
| D | callbacks.py | 2116 profile_batch=2, argument
2132 self._init_profile_batch(profile_batch)
2297 def _init_profile_batch(self, profile_batch): argument
2317 'to profile. Found: {}'.format(profile_batch))
2320 if isinstance(profile_batch, six.string_types):
2321 profile_batch = str(profile_batch).split(',')
2322 profile_batch = nest.map_structure(int, profile_batch)
2324 if isinstance(profile_batch, int):
2325 self._start_batch = profile_batch
2326 self._stop_batch = profile_batch
[all …]
|
| D | callbacks_test.py | 1759 keras.callbacks.TensorBoard(log_dir, profile_batch=0),
2075 profile_batch=0,
2356 profile_batch=0)
2413 self.logdir, histogram_freq=1, profile_batch=1, write_graph=False)
2436 self.logdir, histogram_freq=1, profile_batch=2, write_graph=False)
2459 self.logdir, histogram_freq=1, profile_batch='2,2', write_graph=False)
2483 self.logdir, histogram_freq=1, profile_batch='10,10', write_graph=False)
2519 callbacks=[keras.callbacks.TensorBoard(logdir, profile_batch=1)],
2528 callbacks=[keras.callbacks.TensorBoard(logdir, profile_batch=2)],
2537 self.logdir, histogram_freq=1, profile_batch='1,3', write_graph=False)
[all …]
|
| D | callbacks_v1.py | 130 profile_batch=2): argument
159 self._profile_batch = profile_batch
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.keras.callbacks.-tensor-board.pbtxt | 9 …h\', \'write_images\', \'write_steps_per_second\', \'update_freq\', \'profile_batch\', \'embedding…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.keras.callbacks.-tensor-board.pbtxt | 10 …s\', \'embeddings_metadata\', \'embeddings_data\', \'update_freq\', \'profile_batch\'], varargs=No…
|