- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for SE_InitPlugin (0.2 sec)
-
tensorflow/c/experimental/stream_executor/test/test_pluggable_device.cc
#include "tensorflow/c/experimental/stream_executor/stream_executor.h" #include "tensorflow/c/experimental/stream_executor/stream_executor_test_util.h" extern "C" { void SE_InitPlugin(SE_PlatformRegistrationParams* const params, TF_Status* const status) { stream_executor::test_util::PopulateDefaultPlatformRegistrationParams(params); } void TF_InitKernel() {}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 18 04:53:02 UTC 2021 - 1K bytes - Viewed (0) -
tensorflow/c/experimental/stream_executor/stream_executor.h
// delete_my_device_handle(device->device_handle); // } // // void SE_InitPlugin( // SE_PlatformRegistrationParams* params, // TF_Status* status) { // params->platform = { SP_PLATFORM_STRUCT_SIZE }; // // Values such as `name` and `type` must outlive SE_InitPlugin call. // params->platform->name = DEVICE_NAME; // params->platform->type = DEVICE_TYPE;
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.cc
tensorflow::Env* env = tensorflow::Env::Default(); // Step 1: Load symbol for `TF_InitPlugin` void* dso_symbol; TF_RETURN_IF_ERROR( env->GetSymbolFromLibrary(dso_handle, "SE_InitPlugin", &dso_symbol)); // Step 2: Call `TF_InitPlugin` auto init_fn = reinterpret_cast<SEInitPluginFn>(dso_symbol); return InitStreamExecutorPlugin(init_fn, device_type, platform_name); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 14 07:39:19 UTC 2024 - 27.1K bytes - Viewed (0)