Searched refs:createRnnDescriptor (Results 1 – 8 of 8) sorted by relevance
365 StreamExecutor::createRnnDescriptor( in createRnnDescriptor() function in stream_executor::StreamExecutor377 return dnn_support->createRnnDescriptor( in createRnnDescriptor()
415 port::StatusOr<std::unique_ptr<dnn::RnnDescriptor>> createRnnDescriptor(
2110 createRnnDescriptor(int num_layers, int hidden_size, int input_size, in createRnnDescriptor() function
50 port::StatusOr<std::unique_ptr<dnn::RnnDescriptor>> createRnnDescriptor(
2070 CudnnSupport::createRnnDescriptor( in createRnnDescriptor() function in stream_executor::gpu::CudnnSupport
1041 auto rnn_desc_s = stream->parent()->createRnnDescriptor( in ExtractCudnnRNNParamsInfo()1063 auto rnn_desc_s = executor->createRnnDescriptor( in CreateRnnDescriptor()
84 port::StatusOr<std::unique_ptr<dnn::RnnDescriptor>> createRnnDescriptor(
2515 MIOpenSupport::createRnnDescriptor( in createRnnDescriptor() function in stream_executor::gpu::MIOpenSupport