Home
last modified time | relevance | path

Searched refs:probs_desc (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dctc_loss_op.cc326 std::unique_ptr<RnnStateTensorDescriptor> probs_desc = in Compute() local
352 ->ThenCtcLoss(*probs_desc, probs_data, labels_data, in Compute()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2374 const RnnStateTensorDescriptor& probs_desc, in PrepareForCtcLoss() argument
2384 stream, ToDataType<ElementType>::value, probs_desc, grads_desc, in PrepareForCtcLoss()
2412 const RnnStateTensorDescriptor& probs_desc,
2421 const dnn::RnnStateTensorDescriptor& probs_desc, in DoCtcLoss() argument
2431 DoCtcLoss(stream, ToDataType<ElementType>::value, probs_desc, in DoCtcLoss()
2695 const RnnStateTensorDescriptor& probs_desc, in DoPrepareForCtcLoss() argument
Ddnn.cc625 const RnnStateTensorDescriptor& probs_desc, in DoCtcLoss() argument
Dstream.cc4773 Stream &Stream::ThenCtcLoss(const dnn::RnnStateTensorDescriptor &probs_desc, in ThenCtcLoss() argument
4786 dnn->PrepareForCtcLoss(this, probs_desc, probs_data, grads_desc, in ThenCtcLoss()
4792 status = dnn->DoCtcLoss(this, probs_desc, probs_data, labels_data, in ThenCtcLoss()
Dstream.h1914 Stream &ThenCtcLoss(const dnn::RnnStateTensorDescriptor &probs_desc,
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h563 const dnn::RnnStateTensorDescriptor& probs_desc,
685 Stream* stream, const CudnnRnnStateTensorDescriptor& probs_desc,
708 const dnn::RnnStateTensorDescriptor& probs_desc,
Dcuda_dnn.cc2031 Stream* stream, const CudnnRnnStateTensorDescriptor& probs_desc, in DoCtcLossImpl() argument
2040 int kNumTimestamps = probs_desc.num_layers(); in DoCtcLossImpl()
2041 int kBatchSize = probs_desc.batch_size(); in DoCtcLossImpl()
2042 int kNumLabels = probs_desc.data_size(); in DoCtcLossImpl()
2050 /*handle=*/cudnn.handle(), /*probsDesc=*/probs_desc.handle(), in DoCtcLossImpl()
3887 const dnn::RnnStateTensorDescriptor& probs_desc, in DoPrepareForCtcLoss() argument
3900 static_cast<const CudnnRnnStateTensorDescriptor&>(probs_desc); in DoPrepareForCtcLoss()
3957 const dnn::RnnStateTensorDescriptor& probs_desc, in DoCtcLoss() argument
3972 static_cast<const CudnnRnnStateTensorDescriptor&>(probs_desc); in DoCtcLoss()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h675 const dnn::RnnStateTensorDescriptor& probs_desc,
846 Stream* stream, const MIOpenRnnStateTensorDescriptor& probs_desc,
856 const dnn::RnnStateTensorDescriptor& probs_desc,
Drocm_dnn.cc2392 const dnn::RnnStateTensorDescriptor& probs_desc, in DoPrepareForCtcLoss() argument
2407 static_cast<const MIOpenRnnStateTensorDescriptor&>(probs_desc); in DoPrepareForCtcLoss()
2455 Stream* stream, const MIOpenRnnStateTensorDescriptor& probs_desc, in DoCtcLossImpl() argument
2464 int kNumTimestamps = probs_desc.num_layers(); in DoCtcLossImpl()
2465 int kBatchSize = probs_desc.batch_size(); in DoCtcLossImpl()
2466 int kNumLabels = probs_desc.data_size(); in DoCtcLossImpl()
2471 miopen.handle(), probs_desc.handle(), probs_data.opaque(), in DoCtcLossImpl()
2486 const dnn::RnnStateTensorDescriptor& probs_desc, in DoCtcLoss() argument
2503 static_cast<const MIOpenRnnStateTensorDescriptor&>(probs_desc); in DoCtcLoss()