Searched refs:callback_fn (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/python/framework/ |
| D | op_callbacks.py | 25 def add_op_callback(callback_fn): argument
105 if callback_fn is None:
107 if not callable(callback_fn):
110 "but is not. Received %s" % callback_fn)
112 ctx.add_op_callback(callback_fn)
|
| /external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| D | stream_executor_test.cc | 112 SE_StatusCallbackFn const callback_fn, in host_callback() argument
892 SE_StatusCallbackFn const callback_fn, in TEST_F()
895 callback_fn(callback_arg, status); in TEST_F()
912 SE_StatusCallbackFn const callback_fn, in TEST_F() argument
915 callback_fn(callback_arg, status); in TEST_F()
|
| D | stream_executor.h | 399 SE_StatusCallbackFn callback_fn, void* callback_arg);
|
| /external/libcap/libcap/include/sys/ |
| D | capability.h | 189 int (callback_fn)(void *detail));
|
| /external/llvm/include/llvm/ADT/ |
| D | STLExtras.h | 76 static Ret callback_fn(intptr_t callable, Params ...params) { in callback_fn() function
87 : callback(callback_fn<typename std::remove_reference<Callable>::type>), in callback()
|
| /external/libcap/libcap/ |
| D | cap_proc.c | 765 void cap_launcher_callback(cap_launch_t attr, int (callback_fn)(void *detail)) in cap_launcher_callback()
767 attr->custom_setup_fn = callback_fn; in cap_launcher_callback()
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| D | libtpu.h | 199 void (*callback_fn)(struct TpuStatus*, void* additional_info),
|
| /external/swiftshader/third_party/llvm-subzero/include/llvm/ADT/ |
| D | STLExtras.h | 62 static Ret callback_fn(intptr_t callable, Params ...params) { in callback_fn() function
73 : callback(callback_fn<typename std::remove_reference<Callable>::type>), in callback()
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_executor_c_api.h | 174 SE_StatusCallbackFn callback_fn, void* ctx);
|
| /external/swiftshader/third_party/llvm-10.0/llvm/include/llvm/ADT/ |
| D | STLExtras.h | 107 static Ret callback_fn(intptr_t callable, Params ...params) {
121 : callback(callback_fn<typename std::remove_reference<Callable>::type>),
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tfe_src.cc | 3536 PyObject* callback_fn = PyList_GET_ITEM(op_exec_info.callbacks, i); local
3537 if (!PyCallable_Check(callback_fn)) {
3542 i, callback_fn->ob_type->tp_name)
3547 PyObject_CallObject(callback_fn, callback_args.get());
|
| /external/llvm-project/llvm/include/llvm/ADT/ |
| D | STLExtras.h | 184 static Ret callback_fn(intptr_t callable, Params ...params) {
205 : callback(callback_fn<typename std::remove_reference<Callable>::type>),
|