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.cc562 std::unique_ptr<RnnDescriptor> rnn_desc; member
789 Status DoForward(OpKernelContext* context, const RnnDescriptor& rnn_desc, in DoForward()
870 OpKernelContext* context, const RnnDescriptor& rnn_desc, in DoBackward()
1052 std::unique_ptr<RnnDescriptor>* rnn_desc) { in ExtractCudnnRNNParamsInfo()
1101 std::unique_ptr<RnnDescriptor>* rnn_desc, in CreateRnnDescriptor()
1127 RnnStateCache* cache, RnnDescriptor** rnn_desc, in GetCachedRnnDescriptor()
1169 std::unique_ptr<RnnDescriptor> rnn_desc; in Compute() local
1242 std::unique_ptr<RnnDescriptor> rnn_desc; in Compute() local
1419 std::unique_ptr<RnnDescriptor> rnn_desc; in Compute() local
1752 std::unique_ptr<RnnDescriptor> rnn_desc; in MaybeAutotune() local
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_dnn.cc603 cudnnRNNDescriptor_t rnn_desc, int batch_size, cudnnDataType_t data_type) { in CreatePersistentRnnPlan()
1233 CudnnRnnDescriptor(const CudnnHandle& cudnn, gpu::RnnDescriptor rnn_desc, in CudnnRnnDescriptor()
1274 gpu::RnnDescriptor rnn_desc = CreateRnnDescriptor(); in Create() local
1463 const CudnnHandle& cudnn, cudnnRNNDescriptor_t rnn_desc, const int layer, in CheckAndFetchProjectionWeights()
1535 cudnnRNNDescriptor_t rnn_desc, cudnnRNNMode_t rnn_mode, in Create()
1773 const CudnnRnnDescriptor& rnn_desc, in ExtractAndCheckRnnForward()
1834 const CudnnHandle& cudnn, const CudnnRnnDescriptor& rnn_desc, in CheckRNNParameterSize()
1857 const CudnnRnnDescriptor& rnn_desc, in CreateRnnWorkspace()
1930 Stream* stream, const CudnnRnnDescriptor& rnn_desc, in DoRnnForwardImpl()
2143 Stream* stream, const CudnnRnnDescriptor& rnn_desc, in DoRnnBackwardImpl()
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2027 const MIOpenRnnDescriptor& rnn_desc, in ExtractAndCheckRnnForward()
2087 miopenHandle_t miopen_handle, const MIOpenRnnDescriptor& rnn_desc, in CheckRNNParameterSize()
2103 const MIOpenRnnDescriptor& rnn_desc, in CreateRnnWorkspace()
2137 Stream* stream, const MIOpenRnnDescriptor& rnn_desc, in DoRnnForwardImpl()
2252 Stream* stream, const MIOpenRnnDescriptor& rnn_desc, in DoRnnBackwardImpl()
2370 miopenHandle_t miopen_handle, const MIOpenRnnDescriptor& rnn_desc) in MIOpenRnnParamsDescriptor()
2591 std::unique_ptr<MIOpenRnnDescriptor> rnn_desc(new MIOpenRnnDescriptor( in createRnnDescriptor() local
2632 Stream* stream, const dnn::RnnDescriptor& rnn_desc, in DoRnnForward()
2676 Stream* stream, const dnn::RnnDescriptor& rnn_desc, in DoRnnForward()
2719 Stream* stream, const dnn::RnnDescriptor& rnn_desc, in DoRnnForward()
[all …]
/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Ddnn.h2113 virtual bool DoRnnForward(Stream* stream, const dnn::RnnDescriptor& rnn_desc, in DoRnnForward()
2135 virtual bool DoRnnForward(Stream* stream, const dnn::RnnDescriptor& rnn_desc, in DoRnnForward()
2157 virtual bool DoRnnForward(Stream* stream, const dnn::RnnDescriptor& rnn_desc, in DoRnnForward()
2220 Stream* stream, const dnn::RnnDescriptor& rnn_desc, in DoRnnBackward()
2249 Stream* stream, const dnn::RnnDescriptor& rnn_desc, in DoRnnBackward()
2278 Stream* stream, const dnn::RnnDescriptor& rnn_desc, in DoRnnBackward()
Dstream.cc2031 const dnn::RnnDescriptor &rnn_desc, in ThenRnnForward()
2067 const dnn::RnnDescriptor &rnn_desc, in ThenRnnForward()
2102 const dnn::RnnDescriptor &rnn_desc, in ThenRnnForward()
2138 const dnn::RnnDescriptor &rnn_desc, in ThenRnnBackward()
2184 const dnn::RnnDescriptor &rnn_desc, in ThenRnnBackward()
2229 const dnn::RnnDescriptor &rnn_desc, in ThenRnnBackward()