Home
last modified time | relevance | path

Searched refs:MIOpenSupport (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc487 MIOpenSupport::MIOpenSupport(GpuExecutor* parent) : parent_(parent) {} in MIOpenSupport() function in stream_executor::gpu::MIOpenSupport
489 port::Status MIOpenSupport::Init() { in Init()
519 MIOpenSupport::GetVersion() { in GetVersion()
2001 bool MIOpenSupport::DoRnnForwardImpl( in DoRnnForwardImpl()
2116 bool MIOpenSupport::DoRnnBackwardImpl( in DoRnnBackwardImpl()
2275 MIOpenSupport::createRnnDescriptor( in createRnnDescriptor()
2297 MIOpenSupport::createRnnSequenceTensorDescriptor(int seq_length, int batch_size, in createRnnSequenceTensorDescriptor()
2311 MIOpenSupport::createRnnStateTensorDescriptor(int num_layer, int batch_size, in createRnnStateTensorDescriptor()
2324 bool MIOpenSupport::DoRnnForward( in DoRnnForward()
2367 bool MIOpenSupport::DoRnnForward( in DoRnnForward()
[all …]
Drocm_dnn.h41 class MIOpenSupport : public dnn::DnnSupport {
43 explicit MIOpenSupport(GpuExecutor* parent);
770 SE_DISALLOW_COPY_AND_ASSIGN(MIOpenSupport);