Searched refs:device_fns (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| D | stream_executor.cc | 139 port::Status ValidateSPDeviceFns(const SP_DeviceFns& device_fns) { in ValidateSPDeviceFns() argument
140 VALIDATE_STRUCT_SIZE(SP_DeviceFns, device_fns, SP_DEVICE_FNS_STRUCT_SIZE); in ValidateSPDeviceFns()
239 explicit CStreamExecutor(SP_Device device, SP_DeviceFns* device_fns, in CStreamExecutor() argument
245 device_fns_(device_fns), in CStreamExecutor()
677 SP_DeviceFns device_fns, SP_StreamExecutor stream_executor, in CPlatform() argument
683 device_fns_(std::move(device_fns)), in CPlatform()
780 SP_DeviceFns device_fns{SP_DEVICE_FNS_STRUCT_SIZE}; in InitStreamExecutorPlugin() local
781 device_fns_params.device_fns = &device_fns; in InitStreamExecutorPlugin()
786 TF_RETURN_IF_ERROR(ValidateSPDeviceFns(device_fns)); in InitStreamExecutorPlugin()
809 params.destroy_platform_fns, std::move(device_fns), std::move(se), in InitStreamExecutorPlugin()
|
| D | stream_executor.h | 229 SP_DeviceFns* device_fns; // output, to be filled by plugin member
233 TF_OFFSET_OF_END(SE_CreateDeviceFnsParams, device_fns)
463 SP_DeviceFns* device_fns);
|
| D | stream_executor_internal.h | 54 SP_DeviceFns device_fns, SP_StreamExecutor stream_executor,
|
| D | stream_executor_test.cc | 147 void PopulateDefaultDeviceFns(SP_DeviceFns* device_fns) { in PopulateDefaultDeviceFns() argument
148 *device_fns = {SP_DEVICE_FNS_STRUCT_SIZE}; in PopulateDefaultDeviceFns()
185 params->device_fns->struct_size = {SP_DEVICE_FNS_STRUCT_SIZE}; in create_device_fns()
187 void destroy_device_fns(const SP_Platform* platform, SP_DeviceFns* device_fns) { in destroy_device_fns() argument
|