Searched refs:createRnnSequenceTensorDescriptor (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/stream_executor/ |
D | stream_executor_pimpl.cc | 399 StreamExecutor::createRnnSequenceTensorDescriptor(int max_seq_length, in createRnnSequenceTensorDescriptor() function in stream_executor::StreamExecutor 407 return dnn_support->createRnnSequenceTensorDescriptor( in createRnnSequenceTensorDescriptor() 412 StreamExecutor::createRnnSequenceTensorDescriptor( in createRnnSequenceTensorDescriptor() function in stream_executor::StreamExecutor 421 return dnn_support->createRnnSequenceTensorDescriptor( in createRnnSequenceTensorDescriptor()
|
D | stream_executor_pimpl.h | 417 createRnnSequenceTensorDescriptor(int max_seq_length, int batch_size, 421 createRnnSequenceTensorDescriptor(int max_seq_length, int batch_size,
|
D | dnn.h | 2063 createRnnSequenceTensorDescriptor(int max_seq_length, int batch_size, 2070 createRnnSequenceTensorDescriptor(int max_seq_length, int batch_size,
|
/external/tensorflow/tensorflow/core/kernels/ |
D | cudnn_rnn_ops.cc | 666 auto input_desc_s = executor->createRnnSequenceTensorDescriptor( in CreateForwardAndBackwardIODescriptors() 672 auto input_desc_s = executor->createRnnSequenceTensorDescriptor( in CreateForwardAndBackwardIODescriptors() 679 auto input_desc_s = executor->createRnnSequenceTensorDescriptor( in CreateForwardAndBackwardIODescriptors() 704 auto output_desc_s = executor->createRnnSequenceTensorDescriptor( in CreateForwardAndBackwardIODescriptors() 710 auto output_desc_s = executor->createRnnSequenceTensorDescriptor( in CreateForwardAndBackwardIODescriptors() 717 auto output_desc_s = executor->createRnnSequenceTensorDescriptor( in CreateForwardAndBackwardIODescriptors()
|
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_dnn.h | 58 createRnnSequenceTensorDescriptor(int max_seq_length, int batch_size, 63 createRnnSequenceTensorDescriptor(int max_seq_length, int batch_size,
|
D | cuda_dnn.cc | 1838 CudnnSupport::createRnnSequenceTensorDescriptor(int max_seq_length, in createRnnSequenceTensorDescriptor() function in stream_executor::gpu::CudnnSupport 1850 CudnnSupport::createRnnSequenceTensorDescriptor( in createRnnSequenceTensorDescriptor() function in stream_executor::gpu::CudnnSupport
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_dnn.h | 56 createRnnSequenceTensorDescriptor(int seq_length, int batch_size,
|
D | rocm_dnn.cc | 2297 MIOpenSupport::createRnnSequenceTensorDescriptor(int seq_length, int batch_size, in createRnnSequenceTensorDescriptor() function in stream_executor::gpu::MIOpenSupport
|