Searched defs:seq_lengths_data (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | dnn.h | 2116 const DeviceMemory<int>& seq_lengths_data, in DoRnnForward()
2138 const DeviceMemory<int>& seq_lengths_data, in DoRnnForward()
2160 const DeviceMemory<int>& seq_lengths_data, in DoRnnForward()
2223 const DeviceMemory<int>& seq_lengths_data, in DoRnnBackward()
2252 const DeviceMemory<int>& seq_lengths_data, in DoRnnBackward()
2281 const DeviceMemory<int>& seq_lengths_data, in DoRnnBackward()
|
| H A D | stream.cc | 2034 const DeviceMemory<int> &seq_lengths_data, in ThenRnnForward()
2070 const DeviceMemory<int> &seq_lengths_data, in ThenRnnForward()
2105 const DeviceMemory<int> &seq_lengths_data, in ThenRnnForward()
2141 const DeviceMemory<int> &seq_lengths_data, in ThenRnnBackward()
2187 const DeviceMemory<int> &seq_lengths_data, in ThenRnnBackward()
2232 const DeviceMemory<int> &seq_lengths_data, in ThenRnnBackward()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_dnn.cc | 1933 const DeviceMemory<int>& seq_lengths_data, in DoRnnForwardImpl()
2146 const DeviceMemory<int>& seq_lengths_data, in DoRnnBackwardImpl()
2473 const DeviceMemory<int>& seq_lengths_data, in DoRnnForward()
2517 const DeviceMemory<int>& seq_lengths_data, in DoRnnForward()
2560 const DeviceMemory<int>& seq_lengths_data, in DoRnnForward()
2604 const DeviceMemory<int>& seq_lengths_data, in DoRnnBackward()
2657 const DeviceMemory<int>& seq_lengths_data, in DoRnnBackward()
2709 const DeviceMemory<int>& seq_lengths_data, in DoRnnBackward()
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_dnn.cc | 2635 const DeviceMemory<int>& seq_lengths_data, in DoRnnForward()
2679 const DeviceMemory<int>& seq_lengths_data, in DoRnnForward()
2722 const DeviceMemory<int>& seq_lengths_data, in DoRnnForward()
2745 const DeviceMemory<int>& seq_lengths_data, in DoRnnBackward()
2798 const DeviceMemory<int>& seq_lengths_data, in DoRnnBackward()
2850 const DeviceMemory<int>& seq_lengths_data, in DoRnnBackward()
|