Home
last modified time | relevance | path

Searched defs:rnn_desc (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc559 std::unique_ptr<RnnDescriptor> rnn_desc; member
786 Status DoForward(OpKernelContext* context, const RnnDescriptor& rnn_desc, in DoForward()
847 OpKernelContext* context, const RnnDescriptor& rnn_desc, in DoBackward()
1010 std::unique_ptr<RnnDescriptor>* rnn_desc) { in ExtractCudnnRNNParamsInfo()
1059 std::unique_ptr<RnnDescriptor>* rnn_desc, in CreateRnnDescriptor()
1085 RnnStateCache* cache, RnnDescriptor** rnn_desc, in GetCachedRnnDescriptor()
1127 std::unique_ptr<RnnDescriptor> rnn_desc; in Compute() local
1200 std::unique_ptr<RnnDescriptor> rnn_desc; in Compute() local
1377 std::unique_ptr<RnnDescriptor> rnn_desc; in Compute() local
1708 std::unique_ptr<RnnDescriptor> rnn_desc; in MaybeAutoTune() local
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc506 cudnnRNNDescriptor_t rnn_desc, int batch_size, cudnnDataType_t data_type) { in CreatePersistentRnnPlan()
1058 CudnnRnnDescriptor(const CudnnHandle& cudnn, gpu::RnnDescriptor rnn_desc, in CudnnRnnDescriptor()
1099 gpu::RnnDescriptor rnn_desc = CreateRnnDescriptor(); in Create() local
1285 const CudnnHandle& cudnn, cudnnRNNDescriptor_t rnn_desc, const int layer, in CheckAndFetchProjectionWeights()
1357 cudnnRNNDescriptor_t rnn_desc, cudnnRNNMode_t rnn_mode, in Create()
1598 const CudnnRnnDescriptor& rnn_desc, in ExtractAndCheckRnnForward()
1659 const CudnnHandle& cudnn, const CudnnRnnDescriptor& rnn_desc, in CheckRNNParameterSize()
1676 const CudnnRnnDescriptor& rnn_desc, in CreateRnnWorkspace()
1747 Stream* stream, const CudnnRnnDescriptor& rnn_desc, in DoRnnForwardImpl()
1887 Stream* stream, const CudnnRnnDescriptor& rnn_desc, in DoRnnBackwardImpl()
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1973 const MIOpenRnnDescriptor& rnn_desc, in ExtractAndCheckRnnForward()
2033 miopenHandle_t miopen_handle, const MIOpenRnnDescriptor& rnn_desc, in CheckRNNParameterSize()
2049 const MIOpenRnnDescriptor& rnn_desc, in CreateRnnWorkspace()
2083 Stream* stream, const MIOpenRnnDescriptor& rnn_desc, in DoRnnForwardImpl()
2198 Stream* stream, const MIOpenRnnDescriptor& rnn_desc, in DoRnnBackwardImpl()
2316 miopenHandle_t miopen_handle, const MIOpenRnnDescriptor& rnn_desc) in MIOpenRnnParamsDescriptor()
2537 std::unique_ptr<MIOpenRnnDescriptor> rnn_desc(new MIOpenRnnDescriptor( in createRnnDescriptor() local
2578 Stream* stream, const dnn::RnnDescriptor& rnn_desc, in DoRnnForward()
2621 Stream* stream, const dnn::RnnDescriptor& rnn_desc, in DoRnnForward()
2663 Stream* stream, const dnn::RnnDescriptor& rnn_desc, in DoRnnForward()
[all …]
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2185 virtual bool DoRnnForward(Stream* stream, const dnn::RnnDescriptor& rnn_desc, in DoRnnForward()
2206 virtual bool DoRnnForward(Stream* stream, const dnn::RnnDescriptor& rnn_desc, in DoRnnForward()
2227 virtual bool DoRnnForward(Stream* stream, const dnn::RnnDescriptor& rnn_desc, in DoRnnForward()
2289 Stream* stream, const dnn::RnnDescriptor& rnn_desc, in DoRnnBackward()
2317 Stream* stream, const dnn::RnnDescriptor& rnn_desc, in DoRnnBackward()
2345 Stream* stream, const dnn::RnnDescriptor& rnn_desc, in DoRnnBackward()
Dstream.cc4539 const dnn::RnnDescriptor &rnn_desc, in ThenRnnForward()
4573 const dnn::RnnDescriptor &rnn_desc, in ThenRnnForward()
4606 const dnn::RnnDescriptor &rnn_desc, in ThenRnnForward()
4640 const dnn::RnnDescriptor &rnn_desc, in ThenRnnBackward()
4685 const dnn::RnnDescriptor &rnn_desc, in ThenRnnBackward()
4729 const dnn::RnnDescriptor &rnn_desc, in ThenRnnBackward()