Searched refs:params_handle (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1123 cudnnFilterDescriptor_t params_handle() const { in params_handle() function in stream_executor::gpu::CudnnRnnDescriptor
1574 /*wDesc=*/rnn_desc.params_handle(), /*w=*/params.opaque(), in DoRnnForwardImpl()
1595 /*cx=*/input_c_data.opaque(), /*wDesc=*/rnn_desc.params_handle(), in DoRnnForwardImpl()
1611 /*wDesc=*/rnn_desc.params_handle(), /*w=*/params.opaque(), in DoRnnForwardImpl()
1634 /*cx=*/input_c_data.opaque(), /*wDesc=*/rnn_desc.params_handle(), in DoRnnForwardImpl()
1720 /*wDesc=*/rnn_desc.params_handle(), /*w=*/params.opaque(), in DoRnnBackwardImpl()
1749 /*wDesc=*/rnn_desc.params_handle(), /*w=*/params.opaque(), in DoRnnBackwardImpl()
1776 /*dwDesc=*/rnn_desc.params_handle(), in DoRnnBackwardImpl()
1795 /*dwDesc=*/rnn_desc.params_handle(), in DoRnnBackwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1734 miopenTensorDescriptor_t params_handle() const { in params_handle() function in stream_executor::gpu::MIOpenRnnDescriptor
2080 input_c_data.opaque() /*cx*/, rnn_desc.params_handle() /*wDesc*/, in DoRnnForwardImpl()
2098 input_c_data.opaque() /*cx*/, rnn_desc.params_handle() /*wDesc*/, in DoRnnForwardImpl()
2193 rnn_desc.params_handle() /*wDesc*/, params.opaque() /*w*/, in DoRnnBackwardImpl()
2218 output_data.opaque() /*y*/, rnn_desc.params_handle() /*dwDesc*/, in DoRnnBackwardImpl()
|