Searched defs:rnn_desc (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 555 std::unique_ptr<RnnDescriptor> rnn_desc; member
728 Status DoForward(OpKernelContext* context, const RnnDescriptor& rnn_desc, in DoForward()
788 OpKernelContext* context, const RnnDescriptor& rnn_desc, in DoBackward()
936 std::unique_ptr<RnnDescriptor>* rnn_desc) { in ExtractCudnnRNNParamsInfo()
981 std::unique_ptr<RnnDescriptor>* rnn_desc) { in CreateRnnDescriptor()
1006 RnnDescriptor** rnn_desc) { in GetCachedRnnDescriptor()
1041 std::unique_ptr<RnnDescriptor> rnn_desc; in Compute() local
1085 std::unique_ptr<RnnDescriptor> rnn_desc; in Compute() local
1213 std::unique_ptr<RnnDescriptor> rnn_desc; in Compute() local
1522 std::unique_ptr<RnnDescriptor> rnn_desc; in MaybeAutoTune() local
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 484 cudnnRNNDescriptor_t rnn_desc, int batch_size, cudnnDataType_t data_type) { in CreatePersistentRnnPlan()
1003 CudnnRnnDescriptor(const CudnnHandle& cudnn, gpu::RnnDescriptor rnn_desc, in CudnnRnnDescriptor()
1043 gpu::RnnDescriptor rnn_desc = CreateRnnDescriptor(); in Create() local
1158 cudnnRNNDescriptor_t rnn_desc, cudnnRNNMode_t rnn_mode, in Create()
1412 const CudnnRnnDescriptor& rnn_desc, in ExtractAndCheckRnnForward()
1470 const CudnnHandle& cudnn, const CudnnRnnDescriptor& rnn_desc, in CheckRNNParameterSize()
1487 const CudnnRnnDescriptor& rnn_desc, in CreateRnnWorkspace()
1507 Stream* stream, const CudnnRnnDescriptor& rnn_desc, in DoRnnForwardImpl()
1660 Stream* stream, const CudnnRnnDescriptor& rnn_desc, in DoRnnBackwardImpl()
1872 Stream* stream, const dnn::RnnDescriptor& rnn_desc, in DoRnnForward()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1892 const MIOpenRnnDescriptor& rnn_desc, in ExtractAndCheckRnnForward()
1952 miopenHandle_t miopen_handle, const MIOpenRnnDescriptor& rnn_desc, in CheckRNNParameterSize()
1968 const MIOpenRnnDescriptor& rnn_desc, in CreateRnnWorkspace()
2002 Stream* stream, const MIOpenRnnDescriptor& rnn_desc, in DoRnnForwardImpl()
2117 Stream* stream, const MIOpenRnnDescriptor& rnn_desc, in DoRnnBackwardImpl()
2234 miopenHandle_t miopen_handle, const MIOpenRnnDescriptor& rnn_desc) in MIOpenRnnParamsDescriptor()
2284 std::unique_ptr<MIOpenRnnDescriptor> rnn_desc(new MIOpenRnnDescriptor( in createRnnDescriptor() local
2325 Stream* stream, const dnn::RnnDescriptor& rnn_desc, in DoRnnForward()
2368 Stream* stream, const dnn::RnnDescriptor& rnn_desc, in DoRnnForward()
2410 Stream* stream, const dnn::RnnDescriptor& rnn_desc, in DoRnnForward()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 4846 const dnn::RnnDescriptor &rnn_desc, in ThenRnnForward()
4883 const dnn::RnnDescriptor &rnn_desc, in ThenRnnForward()
4919 const dnn::RnnDescriptor &rnn_desc, in ThenRnnForward()
4956 const dnn::RnnDescriptor &rnn_desc, in ThenRnnBackward()
5003 const dnn::RnnDescriptor &rnn_desc, in ThenRnnBackward()
5049 const dnn::RnnDescriptor &rnn_desc, in ThenRnnBackward()
|