Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2382 int* ctc_loss_algo_id) { in PrepareForCtcLoss() argument
2386 workspace_allocator, scratch_memory, ctc_loss_algo_id); in PrepareForCtcLoss()
2417 DeviceMemory<uint8> scratch_memory, int ctc_loss_algo_id);
2429 DeviceMemory<uint8>* scratch_memory, int ctc_loss_algo_id) { in DoCtcLoss() argument
2434 *scratch_memory, ctc_loss_algo_id), in DoCtcLoss()
2701 int* ctc_loss_algo_id) { in DoPrepareForCtcLoss() argument
Ddnn.cc630 DeviceMemory<uint8> scratch_memory, int ctc_loss_algo_id) { in DoCtcLoss() argument
Dstream.cc4784 int ctc_loss_algo_id; in ThenCtcLoss() local
4789 &scratch_memory, &ctc_loss_algo_id) in ThenCtcLoss()
4795 &scratch_memory, ctc_loss_algo_id); in ThenCtcLoss()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h572 int ctc_loss_algo_id) override;
691 DeviceMemory<uint8> scratch_memory, int ctc_loss_algo_id);
714 int* ctc_loss_algo_id) override;
Dcuda_dnn.cc2037 DeviceMemory<uint8> scratch_memory, int ctc_loss_algo_id) { in DoCtcLossImpl() argument
2048 static_cast<cudnnCTCLossAlgo_t>(ctc_loss_algo_id); in DoCtcLossImpl()
3893 int* ctc_loss_algo_id) { in DoPrepareForCtcLoss() argument
3934 *ctc_loss_algo_id = algo; in DoPrepareForCtcLoss()
3963 int ctc_loss_algo_id) { in DoCtcLoss() argument
3978 scratch_memory, ctc_loss_algo_id); in DoCtcLoss()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h684 int ctc_loss_algo_id) override;
852 DeviceMemory<uint8> scratch_memory, int ctc_loss_algo_id);
862 int* ctc_loss_algo_id) override;
Drocm_dnn.cc2398 int* ctc_loss_algo_id) { in DoPrepareForCtcLoss() argument
2461 DeviceMemory<uint8> scratch_memory, int ctc_loss_algo_id) { in DoCtcLossImpl() argument
2492 int ctc_loss_algo_id) { in DoCtcLoss() argument
2511 scratch_memory, ctc_loss_algo_id); in DoCtcLoss()