Home
last modified time | relevance | path

Searched refs:op_callbacks (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/python/framework/
Dop_callbacks_test.py33 from tensorflow.python.framework import op_callbacks
161 op_callbacks.clear_op_callbacks()
169 op_callbacks.add_op_callback(instrument_0.callback)
170 self.assertEqual(1, len(ctx.op_callbacks))
171 self.assertIn(instrument_0.callback, ctx.op_callbacks)
173 op_callbacks.add_op_callback(instrument_1.callback)
174 self.assertEqual(2, len(ctx.op_callbacks))
175 self.assertIn(instrument_0.callback, ctx.op_callbacks)
176 self.assertIn(instrument_1.callback, ctx.op_callbacks)
178 op_callbacks.remove_op_callback(instrument_1.callback)
[all …]
Dop_callbacks.py126 return ctx.op_callbacks and not ctx.invoking_op_callbacks
141 if ctx.executing_eagerly() and not ctx.op_callbacks:
148 for callback in context.context().op_callbacks:
186 if ctx.op_callbacks:
201 for callback in ctx.op_callbacks:
Dop_def_library.py29 from tensorflow.python.framework import op_callbacks
752 if op_callbacks.should_invoke_op_callbacks():
753 callback_outputs = op_callbacks.invoke_op_callbacks(
/external/tensorflow/tensorflow/python/eager/
Dgraph_only_ops.py24 from tensorflow.python.framework import op_callbacks
42 if op_callbacks.should_invoke_op_callbacks():
45 callback_outputs = op_callbacks.invoke_op_callbacks(
Dcontext.py190 self.op_callbacks = []
1072 if callback not in self._thread_local_data.op_callbacks:
1073 self._thread_local_data.op_callbacks.append(callback)
1084 if callback not in self._thread_local_data.op_callbacks:
1088 del self._thread_local_data.op_callbacks[
1089 self._thread_local_data.op_callbacks.index(callback)]
1092 def op_callbacks(self): member in Context
1093 return self._thread_local_data.op_callbacks
Dexecute.py139 for callback in ctx.op_callbacks:
DBUILD446 "//tensorflow/python:op_callbacks",
Dbenchmarks_test.py81 "MatMul", name, ctx.op_callbacks,
/external/tensorflow/tensorflow/python/debug/lib/
Dcheck_numerics_callback.py29 from tensorflow.python.framework import op_callbacks
417 op_callbacks.add_op_callback(_state.check_numerics_callback.callback)
440 op_callbacks.remove_op_callback(_state.check_numerics_callback.callback)
Ddumping_callback.py39 from tensorflow.python.framework import op_callbacks
782 op_callbacks.add_op_callback(_state.dumping_callback.callback)
811 op_callbacks.remove_op_callback(_state.dumping_callback.callback)
/external/tensorflow/tensorflow/python/distribute/
Dmirrored_strategy.py1014 self._eager_context_op_callbacks = eager_context_state.op_callbacks
1020 eager_context_state.op_callbacks = self._eager_context_op_callbacks
/external/tensorflow/tensorflow/python/debug/
DBUILD79 "//tensorflow/python:op_callbacks",
94 "//tensorflow/python:op_callbacks",
/external/tensorflow/tensorflow/python/
DBUILD1477 ":op_callbacks",
1541 name = "op_callbacks",
1542 srcs = ["framework/op_callbacks.py"],