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