Searched defs:createRnnDescriptor (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/stream_executor/ | ||
D | stream_executor_pimpl.cc | 365 StreamExecutor::createRnnDescriptor( in createRnnDescriptor() function in stream_executor::StreamExecutor |
D | dnn.h | 2110 createRnnDescriptor(int num_layers, int hidden_size, int input_size, in createRnnDescriptor() function |
/external/tensorflow/tensorflow/stream_executor/cuda/ | ||
D | cuda_dnn.cc | 2070 CudnnSupport::createRnnDescriptor( in createRnnDescriptor() function in stream_executor::gpu::CudnnSupport |
/external/tensorflow/tensorflow/stream_executor/rocm/ | ||
D | rocm_dnn.cc | 2515 MIOpenSupport::createRnnDescriptor( in createRnnDescriptor() function in stream_executor::gpu::MIOpenSupport |