Searched refs:params_handle (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1396 cudnnFilterDescriptor_t params_handle() const { in params_handle() function in stream_executor::gpu::CudnnRnnDescriptor
2069 /*wDesc=*/rnn_desc.params_handle(), /*w=*/params.opaque(), in DoRnnForwardImpl()
2085 /*cx=*/input_c_data.opaque(), /*wDesc=*/rnn_desc.params_handle(), in DoRnnForwardImpl()
2099 /*wDesc=*/rnn_desc.params_handle(), /*w=*/params.opaque(), in DoRnnForwardImpl()
2117 /*cx=*/input_c_data.opaque(), /*wDesc=*/rnn_desc.params_handle(), in DoRnnForwardImpl()
2289 /*wDesc=*/rnn_desc.params_handle(), /*w=*/params.opaque(), in DoRnnBackwardImpl()
2313 /*wDesc=*/rnn_desc.params_handle(), /*w=*/params.opaque(), in DoRnnBackwardImpl()
2339 /*dwDesc=*/rnn_desc.params_handle(), in DoRnnBackwardImpl()
2353 /*dwDesc=*/rnn_desc.params_handle(), in DoRnnBackwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1869 miopenTensorDescriptor_t params_handle() const { in params_handle() function in stream_executor::gpu::MIOpenRnnDescriptor
2215 input_c_data.opaque() /*cx*/, rnn_desc.params_handle() /*wDesc*/, in DoRnnForwardImpl()
2233 input_c_data.opaque() /*cx*/, rnn_desc.params_handle() /*wDesc*/, in DoRnnForwardImpl()
2329 rnn_desc.params_handle() /*wDesc*/, params.opaque() /*w*/, in DoRnnBackwardImpl()
2354 output_data.opaque() /*y*/, rnn_desc.params_handle() /*dwDesc*/, in DoRnnBackwardImpl()
|