Searched refs:createRnnDescriptor (Results 1 – 8 of 8) sorted by relevance
381 StreamExecutor::createRnnDescriptor( in createRnnDescriptor() function in stream_executor::StreamExecutor392 return dnn_support->createRnnDescriptor( in createRnnDescriptor()
407 port::StatusOr<std::unique_ptr<dnn::RnnDescriptor>> createRnnDescriptor(
2042 createRnnDescriptor(int num_layers, int hidden_size, int input_size,
964 auto rnn_desc_s = stream->parent()->createRnnDescriptor( in ExtractCudnnRNNParamsInfo()984 auto rnn_desc_s = executor->createRnnDescriptor( in CreateRnnDescriptor()
50 port::StatusOr<std::unique_ptr<dnn::RnnDescriptor>> createRnnDescriptor(
1816 CudnnSupport::createRnnDescriptor( in createRnnDescriptor() function in stream_executor::gpu::CudnnSupport
48 port::StatusOr<std::unique_ptr<dnn::RnnDescriptor>> createRnnDescriptor(
2275 MIOpenSupport::createRnnDescriptor( in createRnnDescriptor() function in stream_executor::gpu::MIOpenSupport