Searched refs:op_callbacks (Results 1 – 18 of 18) sorted by relevance
/external/tensorflow/tensorflow/python/framework/ |
D | op_callbacks_test.py | 32 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 …]
|
D | op_callbacks.py | 126 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:
|
D | constant_op.py | 31 from tensorflow.python.framework import op_callbacks 289 if op_callbacks.should_invoke_op_callbacks(): 292 callback_outputs = op_callbacks.invoke_op_callbacks(
|
D | op_def_library.py | 30 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/ |
D | graph_only_ops.py | 24 from tensorflow.python.framework import op_callbacks 42 if op_callbacks.should_invoke_op_callbacks(): 45 callback_outputs = op_callbacks.invoke_op_callbacks(
|
D | context.py | 1229 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
|
D | pywrap_tfe.h | 420 tensorflow::Safe_PyObjectPtr op_callbacks; member
|
D | execute.py | 139 for callback in ctx.op_callbacks:
|
D | BUILD | 565 "//tensorflow/python:op_callbacks",
|
D | pywrap_tfe_src.cc | 3600 op_exec_info.callbacks = tld->op_callbacks.get(); 4374 thread_local_data->op_callbacks.reset(PyList_New(0));
|
/external/tensorflow/tensorflow/python/autograph/utils/ |
D | testing.py | 27 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/ |
D | check_numerics_callback.py | 30 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)
|
D | dumping_callback.py | 40 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/ |
D | mirrored_run.py | 351 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/ |
D | BUILD | 80 "//tensorflow/python:op_callbacks", 95 "//tensorflow/python:op_callbacks",
|
/external/tensorflow/tensorflow/python/ |
D | tfe_wrapper.cc | 430 return GetPyObject(&GetData()->op_callbacks); in get_op_callbacks() 433 SetPyObject(v, &GetData()->op_callbacks); in set_op_callbacks()
|
D | BUILD | 1017 ":op_callbacks", 1441 name = "op_callbacks", 1442 srcs = ["framework/op_callbacks.py"],
|
/external/tensorflow/tensorflow/python/saved_model/ |
D | load_test.py | 40 from tensorflow.python.framework import op_callbacks 1995 op_callbacks.add_op_callback(_count_restores) 1997 op_callbacks.remove_op_callback(_count_restores)
|