| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | cudnn_rnn_ops.cc | 562 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
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/ |
| H A D | RNN.cpp | 234 RNNDescriptor rnn_desc; in set() local
498 RNNDescriptor rnn_desc; member
555 const RNNDescriptor& rnn_desc, in get_num_weights()
592 const RNNDescriptor& rnn_desc, in add_projection_weights()
700 const RNNDescriptor& rnn_desc, in get_parameters()
878 const RNNDescriptor& rnn_desc, in get_parameters()
1281 RNNDescriptor rnn_desc = rnn.descriptor(handle); in copy_weights_to_flat_buf_views() local
2429 RNNDescriptor rnn_desc = rnn.descriptor(handle); in try_get_weight_buf() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_dnn.cc | 603 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 …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_dnn.cc | 2027 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 …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/ |
| H A D | RNN_miopen.cpp | 127 RNNDescriptor rnn_desc; in descriptor() local
206 RNNDescriptor rnn_desc; member
305 int64_t get_num_weights(miopenHandle_t handle, const RNNDescriptor& rnn_desc, in get_num_weights()
331 … const RNNDescriptor& rnn_desc, const TensorDescriptor& x_desc, const FilterDescriptor& w_desc, in get_parameters()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | dnn.h | 2113 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()
|
| H A D | stream.cc | 2031 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()
|