Searched defs:probs_desc (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ctc_loss_op.cc | 326 std::unique_ptr<RnnStateTensorDescriptor> probs_desc = in Compute() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.cc | 625 const RnnStateTensorDescriptor& probs_desc, in DoCtcLoss()
|
| D | dnn.h | 2374 const RnnStateTensorDescriptor& probs_desc, in PrepareForCtcLoss()
2421 const dnn::RnnStateTensorDescriptor& probs_desc, in DoCtcLoss()
2695 const RnnStateTensorDescriptor& probs_desc, in DoPrepareForCtcLoss()
|
| D | stream.cc | 4773 Stream &Stream::ThenCtcLoss(const dnn::RnnStateTensorDescriptor &probs_desc, in ThenCtcLoss()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 2031 Stream* stream, const CudnnRnnStateTensorDescriptor& probs_desc, in DoCtcLossImpl()
3887 const dnn::RnnStateTensorDescriptor& probs_desc, in DoPrepareForCtcLoss()
3957 const dnn::RnnStateTensorDescriptor& probs_desc, in DoCtcLoss()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 2392 const dnn::RnnStateTensorDescriptor& probs_desc, in DoPrepareForCtcLoss()
2455 Stream* stream, const MIOpenRnnStateTensorDescriptor& probs_desc, in DoCtcLossImpl()
2486 const dnn::RnnStateTensorDescriptor& probs_desc, in DoCtcLoss()
|