Home
last modified time | relevance | path

Searched refs:seq_lengths_data (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_dnn.h84 const DeviceMemory<int>& seq_lengths_data,
103 const DeviceMemory<int>& seq_lengths_data,
122 const DeviceMemory<int>& seq_lengths_data,
141 const DeviceMemory<int>& seq_lengths_data,
167 const DeviceMemory<int>& seq_lengths_data,
193 const DeviceMemory<int>& seq_lengths_data,
571 const DeviceMemory<int>& seq_lengths_data,
591 const DeviceMemory<int>& seq_lengths_data,
Dcuda_dnn.cc1933 const DeviceMemory<int>& seq_lengths_data, in DoRnnForwardImpl() argument
2005 reinterpret_cast<const int*>(seq_lengths_data.opaque()), in DoRnnForwardImpl()
2146 const DeviceMemory<int>& seq_lengths_data, in DoRnnBackwardImpl() argument
2211 reinterpret_cast<const int*>(seq_lengths_data.opaque()), in DoRnnBackwardImpl()
2236 reinterpret_cast<const int*>(seq_lengths_data.opaque()), in DoRnnBackwardImpl()
2473 const DeviceMemory<int>& seq_lengths_data, in DoRnnForward() argument
2505 seq_lengths_data, cudnn_input_h_desc, input_h_data, in DoRnnForward()
2517 const DeviceMemory<int>& seq_lengths_data, in DoRnnForward() argument
2548 seq_lengths_data, cudnn_input_h_desc, input_h_data, in DoRnnForward()
2560 const DeviceMemory<int>& seq_lengths_data, in DoRnnForward() argument
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h104 const DeviceMemory<int>& seq_lengths_data,
123 const DeviceMemory<int>& seq_lengths_data,
142 const DeviceMemory<int>& seq_lengths_data,
161 const DeviceMemory<int>& seq_lengths_data,
187 const DeviceMemory<int>& seq_lengths_data,
213 const DeviceMemory<int>& seq_lengths_data,
Drocm_dnn.cc2635 const DeviceMemory<int>& seq_lengths_data, in DoRnnForward() argument
2679 const DeviceMemory<int>& seq_lengths_data, in DoRnnForward() argument
2722 const DeviceMemory<int>& seq_lengths_data, in DoRnnForward() argument
2745 const DeviceMemory<int>& seq_lengths_data, in DoRnnBackward() argument
2798 const DeviceMemory<int>& seq_lengths_data, in DoRnnBackward() argument
2850 const DeviceMemory<int>& seq_lengths_data, in DoRnnBackward() argument
/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Dstream.cc2034 const DeviceMemory<int> &seq_lengths_data, in ThenRnnForward() argument
2052 this, rnn_desc, input_desc, input_data, seq_lengths_data, input_h_desc, in ThenRnnForward()
2070 const DeviceMemory<int> &seq_lengths_data, in ThenRnnForward() argument
2087 this, rnn_desc, input_desc, input_data, seq_lengths_data, input_h_desc, in ThenRnnForward()
2105 const DeviceMemory<int> &seq_lengths_data, in ThenRnnForward() argument
2123 this, rnn_desc, input_desc, input_data, seq_lengths_data, input_h_desc, in ThenRnnForward()
2141 const DeviceMemory<int> &seq_lengths_data, in ThenRnnBackward() argument
2166 this, rnn_desc, input_desc, input_data, seq_lengths_data, input_h_desc, in ThenRnnBackward()
2187 const DeviceMemory<int> &seq_lengths_data, in ThenRnnBackward() argument
2211 this, rnn_desc, input_desc, input_data, seq_lengths_data, input_h_desc, in ThenRnnBackward()
[all …]
Ddnn.h2116 const DeviceMemory<int>& seq_lengths_data, in DoRnnForward() argument
2138 const DeviceMemory<int>& seq_lengths_data, in DoRnnForward() argument
2160 const DeviceMemory<int>& seq_lengths_data, in DoRnnForward() argument
2223 const DeviceMemory<int>& seq_lengths_data, in DoRnnBackward() argument
2252 const DeviceMemory<int>& seq_lengths_data, in DoRnnBackward() argument
2281 const DeviceMemory<int>& seq_lengths_data, in DoRnnBackward() argument
Dstream.h1229 const DeviceMemory<int> &seq_lengths_data,
1249 const DeviceMemory<int> &seq_lengths_data,
1268 const DeviceMemory<int> &seq_lengths_data,
1290 const DeviceMemory<int> &seq_lengths_data,
1316 const DeviceMemory<int> &seq_lengths_data,
1342 const DeviceMemory<int> &seq_lengths_data,