Home
last modified time | relevance | path

Searched refs:SP_DeviceFns (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/c/experimental/stream_executor/
Dstream_executor.h199 typedef struct SP_DeviceFns { struct
221 } SP_DeviceFns; argument
223 #define SP_DEVICE_FNS_STRUCT_SIZE TF_OFFSET_OF_END(SP_DeviceFns, get_gflops)
229 SP_DeviceFns* device_fns; // output, to be filled by plugin
463 SP_DeviceFns* device_fns);
Dstream_executor_internal.h54 SP_DeviceFns device_fns, SP_StreamExecutor stream_executor,
86 SP_DeviceFns device_fns_;
Dstream_executor.cc139 port::Status ValidateSPDeviceFns(const SP_DeviceFns& device_fns) { in ValidateSPDeviceFns()
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()
663 SP_DeviceFns* device_fns_;
677 SP_DeviceFns device_fns, SP_StreamExecutor stream_executor, in CPlatform()
780 SP_DeviceFns device_fns{SP_DEVICE_FNS_STRUCT_SIZE}; in InitStreamExecutorPlugin()
Dstream_executor_test.cc147 void PopulateDefaultDeviceFns(SP_DeviceFns* device_fns) { in PopulateDefaultDeviceFns()
187 void destroy_device_fns(const SP_Platform* platform, SP_DeviceFns* device_fns) { in destroy_device_fns()
338 SP_DeviceFns device_fns_;