Searched defs:reserve_space_data (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1680 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackwardImpl()
2018 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackward()
2069 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackward()
2121 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackward()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 2137 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackwardImpl()
2453 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackward()
2504 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackward()
2556 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackward()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 4977 DeviceMemory<uint8> *reserve_space_data, in ThenRnnBackward()
5023 DeviceMemory<uint8> *reserve_space_data, in ThenRnnBackward()
5070 DeviceMemory<uint8> *reserve_space_data, in ThenRnnBackward()
|