Home
last modified time | relevance | path

Searched defs:ctc_loss_algo_id (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Ddnn.h2316 int* ctc_loss_algo_id) { in PrepareForCtcLoss()
2363 DeviceMemory<uint8>* scratch_memory, int ctc_loss_algo_id) { in DoCtcLoss()
2641 int* ctc_loss_algo_id) { in DoPrepareForCtcLoss()
H A Ddnn.cc882 DeviceMemory<uint8> scratch_memory, int ctc_loss_algo_id) { in DoCtcLoss()
H A Dstream.cc2285 int ctc_loss_algo_id; in ThenCtcLoss() local
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.cc2452 int* ctc_loss_algo_id) { in DoPrepareForCtcLoss()
2515 DeviceMemory<uint8> scratch_memory, int ctc_loss_algo_id) { in DoCtcLossImpl()
2546 int ctc_loss_algo_id) { in DoCtcLoss()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.cc2380 DeviceMemory<uint8> scratch_memory, int ctc_loss_algo_id) { in DoCtcLossImpl()
5844 int* ctc_loss_algo_id) { in DoPrepareForCtcLoss()
5914 int ctc_loss_algo_id) { in DoCtcLoss()