Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/framework/
Dop_callbacks_test.py28 from tensorflow.python.framework import op_callbacks
157 op_callbacks.clear_op_callbacks()
165 op_callbacks.add_op_callback(instrument_0.callback)
166 self.assertEqual(1, len(ctx.op_callbacks))
167 self.assertIn(instrument_0.callback, ctx.op_callbacks)
169 op_callbacks.add_op_callback(instrument_1.callback)
170 self.assertEqual(2, len(ctx.op_callbacks))
171 self.assertIn(instrument_0.callback, ctx.op_callbacks)
172 self.assertIn(instrument_1.callback, ctx.op_callbacks)
174 op_callbacks.remove_op_callback(instrument_1.callback)
[all …]
Dop_callbacks.py122 return ctx.op_callbacks and not ctx.invoking_op_callbacks
137 if ctx.executing_eagerly() and not ctx.op_callbacks:
144 for callback in context.context().op_callbacks:
182 if ctx.op_callbacks:
197 for callback in ctx.op_callbacks:
Dconstant_op.py27 from tensorflow.python.framework import op_callbacks
292 if op_callbacks.should_invoke_op_callbacks():
295 callback_outputs = op_callbacks.invoke_op_callbacks(
Dop_def_library.py25 from tensorflow.python.framework import op_callbacks
805 if op_callbacks.should_invoke_op_callbacks():
806 callback_outputs = op_callbacks.invoke_op_callbacks(
DBUILD482 ":op_callbacks",
943 name = "op_callbacks",
944 srcs = ["op_callbacks.py"],
/external/tensorflow/tensorflow/python/eager/
Dgraph_only_ops.py20 from tensorflow.python.framework import op_callbacks
38 if op_callbacks.should_invoke_op_callbacks():
41 callback_outputs = op_callbacks.invoke_op_callbacks(
Dcontext.py1371 if callback not in self._thread_local_data.op_callbacks:
1372 self._thread_local_data.op_callbacks.append(callback)
1383 if callback not in self._thread_local_data.op_callbacks:
1386 del self._thread_local_data.op_callbacks[
1387 self._thread_local_data.op_callbacks.index(callback)]
1390 def op_callbacks(self): member in Context
1391 return self._thread_local_data.op_callbacks
Dpywrap_tfe.h418 tensorflow::Safe_PyObjectPtr op_callbacks; member
Dexecute.py128 for callback in ctx.op_callbacks:
DBUILD571 "//tensorflow/python:op_callbacks",
Dpywrap_tfe_src.cc3683 op_exec_info.callbacks = tld->op_callbacks.get();
4202 thread_local_data->op_callbacks.reset(PyList_New(0));
/external/tensorflow/tensorflow/python/autograph/utils/
Dtesting.py23 from tensorflow.python.framework import op_callbacks
121 op_callbacks.add_op_callback(self._op_callback)
124 op_callbacks.remove_op_callback(self._op_callback)
/external/tensorflow/tensorflow/python/debug/lib/
Dcheck_numerics_callback.py26 from tensorflow.python.framework import op_callbacks
435 op_callbacks.add_op_callback(_state.check_numerics_callback.callback)
459 op_callbacks.remove_op_callback(_state.check_numerics_callback.callback)
Ddumping_callback.py34 from tensorflow.python.framework import op_callbacks
856 op_callbacks.add_op_callback(_state.dumping_callback.callback)
887 op_callbacks.remove_op_callback(_state.dumping_callback.callback)
DBUILD24 "//tensorflow/python:op_callbacks",
39 "//tensorflow/python:op_callbacks",
/external/tensorflow/tensorflow/python/distribute/
Dmirrored_run.py414 self._eager_context_op_callbacks = eager_context_state.op_callbacks
420 eager_context_state.op_callbacks = self._eager_context_op_callbacks
/external/tensorflow/tensorflow/python/
Dtfe_wrapper.cc539 return GetPyObject(&GetData()->op_callbacks); in get_op_callbacks()
542 SetPyObject(v, &GetData()->op_callbacks); in set_op_callbacks()
DBUILD4375 name = "op_callbacks",
4376 actual = "//tensorflow/python/framework:op_callbacks",
/external/tensorflow/tensorflow/python/saved_model/
Dload_test.py45 from tensorflow.python.framework import op_callbacks
2141 op_callbacks.add_op_callback(_count_restores)
2143 op_callbacks.remove_op_callback(_count_restores)