Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc555 std::unique_ptr<RnnDescriptor> rnn_desc; member
728 Status DoForward(OpKernelContext* context, const RnnDescriptor& rnn_desc, in DoForward() argument
772 ->ThenRnnForward(rnn_desc, *input_desc, input_data, *state_desc, in DoForward()
788 OpKernelContext* context, const RnnDescriptor& rnn_desc, in DoBackward() argument
850 ->ThenRnnBackward(rnn_desc, *input_desc, input_data, *state_desc, in DoBackward()
936 std::unique_ptr<RnnDescriptor>* rnn_desc) { in ExtractCudnnRNNParamsInfo() argument
971 *rnn_desc = rnn_desc_s.ConsumeValueOrDie(); in ExtractCudnnRNNParamsInfo()
981 std::unique_ptr<RnnDescriptor>* rnn_desc) { in CreateRnnDescriptor() argument
991 *rnn_desc = rnn_desc_s.ConsumeValueOrDie(); in CreateRnnDescriptor()
1006 RnnDescriptor** rnn_desc) { in GetCachedRnnDescriptor() argument
[all …]
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc484 cudnnRNNDescriptor_t rnn_desc, int batch_size, cudnnDataType_t data_type) { in CreatePersistentRnnPlan() argument
487 cudnnCreatePersistentRNNPlan(rnn_desc, batch_size, data_type, &result)); in CreatePersistentRnnPlan()
980 cudnnRNNDescriptor_t rnn_desc, cudnnRNNMode_t rnn_mode,
1003 CudnnRnnDescriptor(const CudnnHandle& cudnn, gpu::RnnDescriptor rnn_desc, in CudnnRnnDescriptor() argument
1013 : rnn_desc_(std::move(rnn_desc)), in CudnnRnnDescriptor()
1043 gpu::RnnDescriptor rnn_desc = CreateRnnDescriptor(); in Create() local
1048 cudnn.handle(), /*rnnDesc=*/rnn_desc.get(), /*hiddenSize=*/hidden_size, in Create()
1059 cudnnSetRNNPaddingMode(rnn_desc.get(), CUDNN_RNN_PADDED_IO_ENABLED)); in Create()
1067 CreatePersistentRnnPlan(rnn_desc.get(), batch_size, data_type); in Create()
1073 cudnnSetPersistentRNNPlan(rnn_desc.get(), rnn_plan.get())); in Create()
[all …]
Dcuda_dnn.h73 bool DoRnnForward(Stream* stream, const dnn::RnnDescriptor& rnn_desc,
91 bool DoRnnForward(Stream* stream, const dnn::RnnDescriptor& rnn_desc,
109 bool DoRnnForward(Stream* stream, const dnn::RnnDescriptor& rnn_desc,
127 bool DoRnnBackward(Stream* stream, const dnn::RnnDescriptor& rnn_desc,
152 bool DoRnnBackward(Stream* stream, const dnn::RnnDescriptor& rnn_desc,
177 bool DoRnnBackward(Stream* stream, const dnn::RnnDescriptor& rnn_desc,
607 Stream* stream, const CudnnRnnDescriptor& rnn_desc,
626 Stream* stream, const CudnnRnnDescriptor& rnn_desc,
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1649 const MIOpenRnnDescriptor& rnn_desc);
1892 const MIOpenRnnDescriptor& rnn_desc, in ExtractAndCheckRnnForward() argument
1906 model_dims->num_layers = rnn_desc.num_layers(); in ExtractAndCheckRnnForward()
1909 model_dims->hidden_size = rnn_desc.hidden_size(); in ExtractAndCheckRnnForward()
1912 (rnn_desc.direction_mode() == miopenRNNbidirection) ? 2 : 1; in ExtractAndCheckRnnForward()
1952 miopenHandle_t miopen_handle, const MIOpenRnnDescriptor& rnn_desc, in CheckRNNParameterSize() argument
1956 miopen_handle /*handle*/, rnn_desc.handle() /*rnnDesc*/, in CheckRNNParameterSize()
1958 rnn_desc.data_type() /*dataType*/); in CheckRNNParameterSize()
1964 rnn_desc.ParamsSizeInBytes(); in CheckRNNParameterSize()
1968 const MIOpenRnnDescriptor& rnn_desc, in CreateRnnWorkspace() argument
[all …]
Drocm_dnn.h64 bool DoRnnForward(Stream* stream, const dnn::RnnDescriptor& rnn_desc,
82 bool DoRnnForward(Stream* stream, const dnn::RnnDescriptor& rnn_desc,
100 bool DoRnnForward(Stream* stream, const dnn::RnnDescriptor& rnn_desc,
118 bool DoRnnBackward(Stream* stream, const dnn::RnnDescriptor& rnn_desc,
143 bool DoRnnBackward(Stream* stream, const dnn::RnnDescriptor& rnn_desc,
168 bool DoRnnBackward(Stream* stream, const dnn::RnnDescriptor& rnn_desc,
663 bool DoRnnForwardImpl(Stream* stream, const MIOpenRnnDescriptor& rnn_desc,
680 bool DoRnnBackwardImpl(Stream* stream, const MIOpenRnnDescriptor& rnn_desc,
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2116 virtual bool DoRnnForward(Stream* stream, const dnn::RnnDescriptor& rnn_desc,
2137 virtual bool DoRnnForward(Stream* stream, const dnn::RnnDescriptor& rnn_desc,
2158 virtual bool DoRnnForward(Stream* stream, const dnn::RnnDescriptor& rnn_desc,
2220 Stream* stream, const dnn::RnnDescriptor& rnn_desc,
2248 Stream* stream, const dnn::RnnDescriptor& rnn_desc,
2276 Stream* stream, const dnn::RnnDescriptor& rnn_desc,
Dstream.h1737 Stream &ThenRnnForward(const dnn::RnnDescriptor &rnn_desc,
1756 Stream &ThenRnnForward(const dnn::RnnDescriptor &rnn_desc,
1774 Stream &ThenRnnForward(const dnn::RnnDescriptor &rnn_desc,
1795 const dnn::RnnDescriptor &rnn_desc,
1820 Stream &ThenRnnBackward(const dnn::RnnDescriptor &rnn_desc,
1845 Stream &ThenRnnBackward(const dnn::RnnDescriptor &rnn_desc,
Dstream.cc4846 const dnn::RnnDescriptor &rnn_desc, in ThenRnnForward() argument
4867 this, rnn_desc, input_desc, input_data, input_h_desc, input_h_data, in ThenRnnForward()
4883 const dnn::RnnDescriptor &rnn_desc, in ThenRnnForward() argument
4903 this, rnn_desc, input_desc, input_data, input_h_desc, input_h_data, in ThenRnnForward()
4919 const dnn::RnnDescriptor &rnn_desc, in ThenRnnForward() argument
4940 this, rnn_desc, input_desc, input_data, input_h_desc, input_h_data, in ThenRnnForward()
4956 const dnn::RnnDescriptor &rnn_desc, in ThenRnnBackward() argument
4984 this, rnn_desc, input_desc, input_data, input_h_desc, input_h_data, in ThenRnnBackward()
5003 const dnn::RnnDescriptor &rnn_desc, in ThenRnnBackward() argument
5030 this, rnn_desc, input_desc, input_data, input_h_desc, input_h_data, in ThenRnnBackward()
[all …]