- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for SE_InitPlugin (0.21 sec)
-
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)