Home
last modified time | relevance | path

Searched refs:SP_Stream (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/c/experimental/stream_executor/
Dstream_executor.h93 typedef struct SP_Stream_st* SP_Stream; typedef
285 void (*create_stream)(const SP_Device* device, SP_Stream* stream,
289 void (*destroy_stream)(const SP_Device* device, SP_Stream stream);
293 void (*create_stream_dependency)(const SP_Device* device, SP_Stream dependent,
294 SP_Stream other, TF_Status* status);
297 void (*get_stream_status)(const SP_Device* device, SP_Stream stream,
313 void (*record_event)(const SP_Device* device, SP_Stream stream,
317 void (*wait_for_event)(const SP_Device* const device, SP_Stream stream,
331 void (*start_timer)(const SP_Device* device, SP_Stream stream, SP_Timer timer,
335 void (*stop_timer)(const SP_Device* device, SP_Stream stream, SP_Timer timer,
[all …]
Dstream_executor_test.cc65 void create_stream(const SP_Device* const device, SP_Stream* stream, in create_stream()
69 void destroy_stream(const SP_Device* const device, SP_Stream stream) {} in destroy_stream()
71 SP_Stream dependent, SP_Stream other, in create_stream_dependency()
73 void get_stream_status(const SP_Device* const device, SP_Stream stream, in get_stream_status()
83 void record_event(const SP_Device* const device, SP_Stream stream, in record_event()
85 void wait_for_event(const SP_Device* const device, SP_Stream stream, in wait_for_event()
90 void start_timer(const SP_Device* const device, SP_Stream stream, in start_timer()
92 void stop_timer(const SP_Device* const device, SP_Stream stream, SP_Timer timer, in stop_timer()
94 void memcpy_dtoh(const SP_Device* const device, SP_Stream stream, in memcpy_dtoh()
97 void memcpy_htod(const SP_Device* const device, SP_Stream stream, in memcpy_htod()
[all …]
Dstream_executor.cc398 SP_Stream stream_handle = in Memcpy()
412 SP_Stream stream_handle = in Memcpy()
427 SP_Stream stream_handle = in MemcpyDeviceToDevice()
441 SP_Stream stream_handle = in HostCallback()
456 SP_Stream stream_handle = in RecordEvent()
461 SP_Stream stream_handle = in WaitForEvent()
491 SP_Stream dependent_handle = in CreateStreamDependency()
493 SP_Stream other_handle = in CreateStreamDependency()
515 SP_Stream stream_handle = in StartTimer()
529 SP_Stream stream_handle = in StopTimer()
[all …]
Dstream_executor_internal.h116 SP_Stream Handle() { return stream_handle_; } in Handle()
121 SP_Stream stream_handle_;
138 port::Status Record(SP_Stream stream_handle) { in Record()
/external/tensorflow/tensorflow/c/
Dkernels.h144 TF_CAPI_EXPORT extern SP_Stream TF_GetStream(TF_OpKernelContext* ctx,
Dkernels.cc197 SP_Stream TF_GetStream(TF_OpKernelContext* ctx, TF_Status* status) { in TF_GetStream()
Dkernels_test.cc716 SP_Stream stream = TF_GetStream(ctx, s); in TEST_F()
/external/tensorflow/tensorflow/core/
DBUILD878 # Including just the header for SP_Stream reference in kernels C API.