Searched refs:use_padded_io (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 1060 bool use_padded_io) { in CreateRnnDescriptor() argument
1068 use_padded_io); in CreateRnnDescriptor()
1086 bool use_padded_io) { in GetCachedRnnDescriptor() argument
1095 dropout_state_allocator, &rnn_state.rnn_desc, use_padded_io); in GetCachedRnnDescriptor()
1464 bool use_padded_io = false; in ComputeAndReturnAlgorithm() local
1470 use_padded_io = in ComputeAndReturnAlgorithm()
1514 &rnn_state_cache_, &rnn_desc_ptr, use_padded_io)); in ComputeAndReturnAlgorithm()
1864 bool use_padded_io = false; in ComputeImpl() local
1870 use_padded_io = in ComputeImpl()
1917 &rnn_desc_ptr, use_padded_io)); in ComputeImpl()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 371 bool use_padded_io) { in createRnnDescriptor() argument
380 state_allocator, use_padded_io); in createRnnDescriptor()
|
| D | stream_executor_pimpl.h | 421 bool use_padded_io);
|
| D | dnn.h | 2117 ScratchAllocator* state_allocator, bool use_padded_io) { in createRnnDescriptor() argument
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1094 ScratchAllocator* state_allocator, bool use_padded_io) { in Create() argument
1137 if (use_padded_io) aux_flags |= CUDNN_RNN_PADDED_IO_ENABLED; in Create()
1167 if (use_padded_io) { in Create()
2076 bool use_padded_io) { in createRnnDescriptor() argument
2087 algorithm_config, dropout, seed, state_allocator, use_padded_io)); in createRnnDescriptor()
|
| D | cuda_dnn.h | 56 bool use_padded_io) override;
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 90 bool use_padded_io) override;
|
| D | rocm_dnn.cc | 2521 bool use_padded_io) { in createRnnDescriptor() argument
2524 if (use_padded_io) { in createRnnDescriptor()
|