Home
last modified time | relevance | path

Searched refs:reserve_space_data (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h148 DeviceMemory<uint8>* reserve_space_data,
173 DeviceMemory<uint8>* reserve_space_data,
198 DeviceMemory<uint8>* reserve_space_data,
646 DeviceMemory<uint8>* reserve_space_data,
Dcuda_dnn.cc1680 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackwardImpl() argument
1731 /*reserveSpace=*/reserve_space_data->opaque(), in DoRnnBackwardImpl()
1732 /*reserveSpaceSizeInBytes=*/reserve_space_data->size())); in DoRnnBackwardImpl()
1759 /*reserveSpace=*/reserve_space_data->opaque(), in DoRnnBackwardImpl()
1760 /*reserveSpaceSizeInBytes=*/reserve_space_data->size())); in DoRnnBackwardImpl()
1778 /*reserveSpace=*/reserve_space_data->opaque(), in DoRnnBackwardImpl()
1779 /*reserveSpaceSizeInBytes=*/reserve_space_data->size())); in DoRnnBackwardImpl()
1797 /*reserveSpace=*/reserve_space_data->opaque(), in DoRnnBackwardImpl()
1798 /*reserveSpaceSizeInBytes=*/reserve_space_data->size())); in DoRnnBackwardImpl()
2018 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackward() argument
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h139 DeviceMemory<uint8>* reserve_space_data,
164 DeviceMemory<uint8>* reserve_space_data,
189 DeviceMemory<uint8>* reserve_space_data,
701 DeviceMemory<uint8>* reserve_space_data,
Drocm_dnn.cc2137 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackwardImpl() argument
2202 reserve_space_data->opaque() /*reserveSpace*/, in DoRnnBackwardImpl()
2203 reserve_space_data->size() /*reserveSpaceSizeInBytes*/); in DoRnnBackwardImpl()
2221 reserve_space_data->opaque() /*reserveSpace*/, in DoRnnBackwardImpl()
2222 reserve_space_data->size() /*reserveSpaceSizeInBytes*/); in DoRnnBackwardImpl()
2453 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackward() argument
2480 reserve_space_data, workspace_allocator); in DoRnnBackward()
2504 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackward() argument
2531 reserve_space_data, workspace_allocator); in DoRnnBackward()
2556 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackward() argument
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2241 DeviceMemory<uint8>* reserve_space_data,
2269 DeviceMemory<uint8>* reserve_space_data,
2297 DeviceMemory<uint8>* reserve_space_data,
Dstream.h1816 DeviceMemory<uint8> *reserve_space_data,
1841 DeviceMemory<uint8> *reserve_space_data,
1866 DeviceMemory<uint8> *reserve_space_data,
Dstream.cc4977 DeviceMemory<uint8> *reserve_space_data, in ThenRnnBackward() argument
4989 params_backprop_data, reserve_space_data, workspace_allocator, in ThenRnnBackward()
5023 DeviceMemory<uint8> *reserve_space_data, in ThenRnnBackward() argument
5035 params_backprop_data, reserve_space_data, workspace_allocator, in ThenRnnBackward()
5070 DeviceMemory<uint8> *reserve_space_data, in ThenRnnBackward() argument
5082 params_backprop_data, reserve_space_data, workspace_allocator, in ThenRnnBackward()