Searched refs:SE_PlatformRegistrationParams (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| D | stream_executor.h | 485 typedef struct SE_PlatformRegistrationParams { struct
501 } SE_PlatformRegistrationParams; argument
504 TF_OFFSET_OF_END(SE_PlatformRegistrationParams, destroy_platform_fns)
506 void SE_InitPlugin(SE_PlatformRegistrationParams* params, TF_Status* status);
|
| D | stream_executor_test.cc | 211 auto plugin_init = [](SE_PlatformRegistrationParams* const params, in TEST()
233 auto plugin_init = [](SE_PlatformRegistrationParams* const params, in TEST()
248 auto plugin_init = [](SE_PlatformRegistrationParams* const params, in TEST()
265 auto plugin_init = [](SE_PlatformRegistrationParams* const params, in TEST()
281 auto plugin_init = [](SE_PlatformRegistrationParams* const params, in TEST()
297 auto plugin_init = [](SE_PlatformRegistrationParams* const params, in TEST()
|
| D | stream_executor.cc | 182 const SE_PlatformRegistrationParams& params) { in ValidateSEPlatformRegistrationParams()
183 VALIDATE_STRUCT_SIZE(SE_PlatformRegistrationParams, params, in ValidateSEPlatformRegistrationParams()
185 VALIDATE_MEMBER(SE_PlatformRegistrationParams, params, destroy_platform); in ValidateSEPlatformRegistrationParams()
186 VALIDATE_MEMBER(SE_PlatformRegistrationParams, params, destroy_platform_fns); in ValidateSEPlatformRegistrationParams()
760 SE_PlatformRegistrationParams params{ in InitStreamExecutorPlugin()
|
| D | stream_executor_internal.h | 31 typedef void (*SEInitPluginFn)(SE_PlatformRegistrationParams* const,
|
| /external/tensorflow/tensorflow/c/experimental/stream_executor/test/ |
| D | test_pluggable_device.cc | 18 void SE_InitPlugin(SE_PlatformRegistrationParams* const params, in SE_InitPlugin()
|