Home
last modified time | relevance | path

Searched refs:createRnnDescriptor (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_pimpl.cc381 StreamExecutor::createRnnDescriptor( in createRnnDescriptor() function in stream_executor::StreamExecutor
392 return dnn_support->createRnnDescriptor( in createRnnDescriptor()
Dstream_executor_pimpl.h407 port::StatusOr<std::unique_ptr<dnn::RnnDescriptor>> createRnnDescriptor(
Ddnn.h2042 createRnnDescriptor(int num_layers, int hidden_size, int input_size,
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc964 auto rnn_desc_s = stream->parent()->createRnnDescriptor( in ExtractCudnnRNNParamsInfo()
984 auto rnn_desc_s = executor->createRnnDescriptor( in CreateRnnDescriptor()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h50 port::StatusOr<std::unique_ptr<dnn::RnnDescriptor>> createRnnDescriptor(
Dcuda_dnn.cc1816 CudnnSupport::createRnnDescriptor( in createRnnDescriptor() function in stream_executor::gpu::CudnnSupport
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h48 port::StatusOr<std::unique_ptr<dnn::RnnDescriptor>> createRnnDescriptor(
Drocm_dnn.cc2275 MIOpenSupport::createRnnDescriptor( in createRnnDescriptor() function in stream_executor::gpu::MIOpenSupport