- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for host_callback (0.25 sec)
-
tensorflow/c/experimental/stream_executor/stream_executor.cc
return false; } return true; } bool HostCallback(Stream* stream, absl::AnyInvocable<absl::Status() &&> callback) override { SP_Stream stream_handle = static_cast<CStream*>(stream)->Handle(); HostCallbackContext* ctx = new HostCallbackContext{std::move(callback)}; return stream_executor_->host_callback(&device_, stream_handle,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 14 07:39:19 UTC 2024 - 27.1K bytes - Viewed (0) -
tensorflow/c/experimental/stream_executor/stream_executor.h
// `callback_arg` should be passed as the first argument to `callback_fn`. TF_Bool (*host_callback)(const SP_Device* device, SP_Stream stream, SE_StatusCallbackFn callback_fn, void* callback_arg); } SP_StreamExecutor; #define SP_STREAMEXECUTOR_STRUCT_SIZE \ TF_OFFSET_OF_END(SP_StreamExecutor, host_callback) typedef struct SE_CreateStreamExecutorParams { size_t struct_size;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 24 08:40:35 UTC 2022 - 21.6K bytes - Viewed (0) -
tensorflow/c/experimental/stream_executor/stream_executor_test_util.cc
se->sync_memcpy_dtoh = SyncMemcpyDToH; se->sync_memcpy_htod = SyncMemcpyHToD; se->block_host_for_event = BlockHostForEvent; se->synchronize_all_activity = SynchronizeAllActivity; se->host_callback = HostCallback; se->mem_zero = MemZero; se->memset = Memset; se->memset32 = Memset32; } void PopulateDefaultDeviceFns(SP_DeviceFns* device_fns) { *device_fns = {SP_DEVICE_FNS_STRUCT_SIZE}; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 21 20:09:00 UTC 2023 - 8.7K bytes - Viewed (0) -
tensorflow/c/experimental/stream_executor/stream_executor_test.cc
ASSERT_FALSE(synchronize_all_called); ASSERT_TRUE(executor->SynchronizeAllActivity()); ASSERT_TRUE(synchronize_all_called); } TEST_F(StreamExecutorTest, HostCallbackOk) { se_.host_callback = [](const SP_Device* const device, SP_Stream stream, SE_StatusCallbackFn const callback_fn, void* const callback_arg) -> TF_Bool { TF_Status* status = TF_NewStatus();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 19:54:04 UTC 2024 - 26.5K bytes - Viewed (0) -
tests/test_generate_unique_id_function.py
"/post-callback", response_model=List[Item], responses={404: {"model": List[Message]}}, generate_unique_id_function=custom_generate_unique_id3, ) def post_callback(item1: Item, item2: Item): return item1, item2 # pragma: nocover @app.post( "/", response_model=List[Item], responses={404: {"model": List[Message]}},
Registered: Mon Jun 17 08:32:26 UTC 2024 - Last Modified: Sat Jan 13 15:10:26 UTC 2024 - 66.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/mark_ops_for_outside_compilation.mlir
// CHECK: tf.XlaHostCompute // CHECK-SAME:_xla_original_oc_node_name = "hcb0", _xla_token_input_nodes = ["_xla_token_arg_node"] "tf.XlaHostCompute"(%cst) <{ancestors = [], cost_estimate_ns = 1000000 : i64, key = "_host_callback", recv_key = "", send_key = "", shapes = [], tpu_core = 0 : i64}> {_xla_original_oc_node_name = "hcb0", _xla_token_input_nodes = ["_xla_token_arg_node"]} : (tensor<i32>) -> () tf_device.return
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 16:22:32 UTC 2024 - 29.5K bytes - Viewed (0) -
src/internal/trace/traceviewer/static/trace_viewer_full.html
function createButton(label,opt_callback,opt_this){const buttonEl=document.createElement('input');buttonEl.type='button';buttonEl.value=label;function onClick(){opt_callback.call(opt_this||buttonEl);} if(opt_callback){buttonEl.addEventListener('click',onClick);} return buttonEl;}
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Tue Nov 21 20:45:06 UTC 2023 - 2.5M bytes - Viewed (2)