Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dctc_loss_op.cc332 std::unique_ptr<RnnStateTensorDescriptor> grads_desc = in Compute() local
354 *grads_desc, &grads_data, &workspace_allocator) in Compute()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2376 const RnnStateTensorDescriptor& grads_desc, in PrepareForCtcLoss() argument
2384 stream, ToDataType<ElementType>::value, probs_desc, grads_desc, in PrepareForCtcLoss()
2416 const RnnStateTensorDescriptor& grads_desc, DeviceMemoryBase grads_data,
2427 const dnn::RnnStateTensorDescriptor& grads_desc, in DoCtcLoss() argument
2433 input_lengths_data, *costs_data, grads_desc, *grads_data, in DoCtcLoss()
2696 const RnnStateTensorDescriptor& grads_desc, in DoPrepareForCtcLoss() argument
Ddnn.cc629 const RnnStateTensorDescriptor& grads_desc, DeviceMemoryBase grads_data, in DoCtcLoss() argument
Dstream.cc4779 const dnn::RnnStateTensorDescriptor &grads_desc, in ThenCtcLoss() argument
4786 dnn->PrepareForCtcLoss(this, probs_desc, probs_data, grads_desc, in ThenCtcLoss()
4794 costs_data, grads_desc, grads_data, in ThenCtcLoss()
Dstream.h1920 const dnn::RnnStateTensorDescriptor &grads_desc,
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h569 const dnn::RnnStateTensorDescriptor& grads_desc,
689 const CudnnRnnStateTensorDescriptor& grads_desc,
709 const dnn::RnnStateTensorDescriptor& grads_desc,
Dcuda_dnn.cc2035 const CudnnRnnStateTensorDescriptor& grads_desc, in DoCtcLossImpl() argument
2054 /*costs=*/costs_data.opaque(), /*gradientsDesc=*/grads_desc.handle(), in DoCtcLossImpl()
3888 const dnn::RnnStateTensorDescriptor& grads_desc, in DoPrepareForCtcLoss() argument
3902 static_cast<const CudnnRnnStateTensorDescriptor&>(grads_desc); in DoPrepareForCtcLoss()
3961 const dnn::RnnStateTensorDescriptor& grads_desc, in DoCtcLoss() argument
3974 static_cast<const CudnnRnnStateTensorDescriptor&>(grads_desc); in DoCtcLoss()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h681 const dnn::RnnStateTensorDescriptor& grads_desc,
850 const MIOpenRnnStateTensorDescriptor& grads_desc,
857 const dnn::RnnStateTensorDescriptor& grads_desc,
Drocm_dnn.cc2393 const dnn::RnnStateTensorDescriptor& grads_desc, in DoPrepareForCtcLoss() argument
2410 static_cast<const MIOpenRnnStateTensorDescriptor&>(grads_desc); in DoPrepareForCtcLoss()
2459 const MIOpenRnnStateTensorDescriptor& grads_desc, in DoCtcLossImpl() argument
2473 costs_data.opaque(), grads_desc.handle(), grads_data.opaque(), in DoCtcLossImpl()
2490 const dnn::RnnStateTensorDescriptor& grads_desc, in DoCtcLoss() argument
2506 static_cast<const MIOpenRnnStateTensorDescriptor&>(grads_desc); in DoCtcLoss()