Home
last modified time | relevance | path

Searched refs:ctc_loss_desc (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h690 DeviceMemoryBase grads_data, const CudnnCtcLossDescriptor& ctc_loss_desc,
Dcuda_dnn.cc2036 DeviceMemoryBase grads_data, const CudnnCtcLossDescriptor& ctc_loss_desc, in DoCtcLossImpl() argument
2057 /*ctcLossDesc=*/ctc_loss_desc.handle(), in DoCtcLossImpl()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h851 DeviceMemoryBase grads_data, const MIOpenCTCLossDescriptor& ctc_loss_desc,
Drocm_dnn.cc2460 DeviceMemoryBase grads_data, const MIOpenCTCLossDescriptor& ctc_loss_desc, in DoCtcLossImpl() argument
2474 MIOPEN_CTC_LOSS_ALGO_DETERMINISTIC, ctc_loss_desc.handle(), in DoCtcLossImpl()