Searched refs:state_allocator (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 386 ScratchAllocator *state_allocator) { in createRnnDescriptor() argument
395 state_allocator); in createRnnDescriptor()
|
| D | stream_executor_pimpl.h | 412 ScratchAllocator *state_allocator);
|
| D | dnn.h | 2048 ScratchAllocator* state_allocator) {
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 935 ScratchAllocator* state_allocator) { in Create() argument
944 if (state_allocator) { in Create()
948 SE_ASSIGN_OR_RETURN(state_memory, state_allocator->AllocateBytes( in Create()
1038 ScratchAllocator* state_allocator) { in Create() argument
1041 CudnnDropoutDescriptor::Create(cudnn, dropout, seed, state_allocator)); in Create()
1821 ScratchAllocator* state_allocator) { in createRnnDescriptor() argument
1832 algorithm_config, dropout, seed, state_allocator)); in createRnnDescriptor()
|
| D | cuda_dnn.h | 55 ScratchAllocator* state_allocator) override;
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 53 ScratchAllocator* state_allocator) override;
|
| D | rocm_dnn.cc | 1687 ScratchAllocator* state_allocator) in MIOpenRnnDescriptor() argument
2280 ScratchAllocator* state_allocator) { in createRnnDescriptor() argument
2288 ToMIOpenDataType(data_type), dropout, seed, state_allocator)); in createRnnDescriptor()
|