Searched refs:ctc_loss_desc (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_dnn.h | 690 DeviceMemoryBase grads_data, const CudnnCtcLossDescriptor& ctc_loss_desc,
|
D | cuda_dnn.cc | 2036 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/ |
D | rocm_dnn.h | 851 DeviceMemoryBase grads_data, const MIOpenCTCLossDescriptor& ctc_loss_desc,
|
D | rocm_dnn.cc | 2460 DeviceMemoryBase grads_data, const MIOpenCTCLossDescriptor& ctc_loss_desc, in DoCtcLossImpl() argument 2474 MIOPEN_CTC_LOSS_ALGO_DETERMINISTIC, ctc_loss_desc.handle(), in DoCtcLossImpl()
|