Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_executor_c_api.h60 bool TpuExecutor_AllocateStream(SE_StreamExecutor* executor, SE_Stream* stream);
62 SE_Stream* stream);
64 SE_Stream* dependent, SE_Stream* other);
65 void TpuExecutor_GetStatus(SE_StreamExecutor* executor, SE_Stream* stream,
76 void TpuExecutor_RecordEvent(SE_StreamExecutor* executor, SE_Stream* stream,
78 void TpuExecutor_WaitForEvent(SE_StreamExecutor* executor, SE_Stream* stream,
83 bool TpuExecutor_StartTimer(SE_StreamExecutor* executor, SE_Stream* stream,
85 bool TpuExecutor_StopTimer(SE_StreamExecutor* executor, SE_Stream* stream,
96 bool TpuExecutor_MemcpyToHost(SE_StreamExecutor* executor, SE_Stream* stream,
101 bool TpuExecutor_MemcpyFromHost(SE_StreamExecutor* executor, SE_Stream* stream,
[all …]
Dtpu_stream.h33 explicit TpuStream(SE_Stream* stream) : stream_(stream) {} in TpuStream()
76 SE_Stream* se_stream() const { return stream_; } in se_stream()
79 mutable SE_Stream* stream_;
Dc_api_decl.h55 typedef struct SE_Stream SE_Stream; typedef
165 SE_Stream* stream;
166 SE_Stream* host_to_device_stream;
Dc_api_defn.h42 struct SE_Stream { struct
43 explicit SE_Stream(stream_executor::StreamExecutor* parent) in SE_Stream() argument
Dtpu_platform.h37 SE_Stream*>;
124 SE_Stream* LookupStream(stream_executor::internal::StreamInterface* key) { in LookupStream()
Dtpu_executor.h231 SE_Stream* get_stream(StreamInterface* ptr) { in get_stream()
Dtpu_executor.cc178 SE_Stream* tpu_stream = tpu::ExecutorApiFn()->TpuStream_NewFn(executor_); in GetStreamImplementation()
/external/tensorflow/tensorflow/core/tpu/
Dtpu_ops_c_api.h166 SE_Stream* stream;