Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/framework/
Dop_callbacks_test.py32 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:
Dconstant_op.py31 from tensorflow.python.framework import op_callbacks
289 if op_callbacks.should_invoke_op_callbacks():
292 callback_outputs = op_callbacks.invoke_op_callbacks(
Dop_def_library.py30 from tensorflow.python.framework import op_callbacks
758 if op_callbacks.should_invoke_op_callbacks():
759 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.py1229 if callback not in self._thread_local_data.op_callbacks:
1230 self._thread_local_data.op_callbacks.append(callback)
1241 if callback not in self._thread_local_data.op_callbacks:
1245 del self._thread_local_data.op_callbacks[
1246 self._thread_local_data.op_callbacks.index(callback)]
1249 def op_callbacks(self): member in Context
1250 return self._thread_local_data.op_callbacks
Dpywrap_tfe.h420 tensorflow::Safe_PyObjectPtr op_callbacks; member
Dexecute.py139 for callback in ctx.op_callbacks:
DBUILD565 "//tensorflow/python:op_callbacks",
Dpywrap_tfe_src.cc3600 op_exec_info.callbacks = tld->op_callbacks.get();
4374 thread_local_data->op_callbacks.reset(PyList_New(0));
/external/tensorflow/tensorflow/python/autograph/utils/
Dtesting.py27 from tensorflow.python.framework import op_callbacks
125 op_callbacks.add_op_callback(self._op_callback)
128 op_callbacks.remove_op_callback(self._op_callback)
/external/tensorflow/tensorflow/python/debug/lib/
Dcheck_numerics_callback.py30 from tensorflow.python.framework import op_callbacks
439 op_callbacks.add_op_callback(_state.check_numerics_callback.callback)
463 op_callbacks.remove_op_callback(_state.check_numerics_callback.callback)
Ddumping_callback.py40 from tensorflow.python.framework import op_callbacks
865 op_callbacks.add_op_callback(_state.dumping_callback.callback)
896 op_callbacks.remove_op_callback(_state.dumping_callback.callback)
/external/tensorflow/tensorflow/python/distribute/
Dmirrored_run.py351 self._eager_context_op_callbacks = eager_context_state.op_callbacks
357 eager_context_state.op_callbacks = self._eager_context_op_callbacks
/external/tensorflow/tensorflow/python/debug/
DBUILD80 "//tensorflow/python:op_callbacks",
95 "//tensorflow/python:op_callbacks",
/external/tensorflow/tensorflow/python/
Dtfe_wrapper.cc430 return GetPyObject(&GetData()->op_callbacks); in get_op_callbacks()
433 SetPyObject(v, &GetData()->op_callbacks); in set_op_callbacks()
DBUILD1017 ":op_callbacks",
1441 name = "op_callbacks",
1442 srcs = ["framework/op_callbacks.py"],
/external/tensorflow/tensorflow/python/saved_model/
Dload_test.py40 from tensorflow.python.framework import op_callbacks
1995 op_callbacks.add_op_callback(_count_restores)
1997 op_callbacks.remove_op_callback(_count_restores)