Home
last modified time | relevance | path

Searched refs:createRnnSequenceTensorDescriptor (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_pimpl.cc399 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()
Dstream_executor_pimpl.h417 createRnnSequenceTensorDescriptor(int max_seq_length, int batch_size,
421 createRnnSequenceTensorDescriptor(int max_seq_length, int batch_size,
Ddnn.h2063 createRnnSequenceTensorDescriptor(int max_seq_length, int batch_size,
2070 createRnnSequenceTensorDescriptor(int max_seq_length, int batch_size,
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc666 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/
Dcuda_dnn.h58 createRnnSequenceTensorDescriptor(int max_seq_length, int batch_size,
63 createRnnSequenceTensorDescriptor(int max_seq_length, int batch_size,
Dcuda_dnn.cc1838 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/
Drocm_dnn.h56 createRnnSequenceTensorDescriptor(int seq_length, int batch_size,
Drocm_dnn.cc2297 MIOpenSupport::createRnnSequenceTensorDescriptor(int seq_length, int batch_size, in createRnnSequenceTensorDescriptor() function in stream_executor::gpu::MIOpenSupport