Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h149 DeviceMemory<uint8>* reserve_space_data,
174 DeviceMemory<uint8>* reserve_space_data,
199 DeviceMemory<uint8>* reserve_space_data,
254 DeviceMemory<uint8>* reserve_space_data,
265 DeviceMemory<uint8>* reserve_space_data,
611 DeviceMemory<U>* offset_backprop, DeviceMemory<uint8>* reserve_space_data,
680 DeviceMemory<uint8>* reserve_space_data,
Dcuda_dnn.cc1907 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackwardImpl() argument
1957 /*reserveSpace=*/reserve_space_data->opaque(), in DoRnnBackwardImpl()
1958 /*reserveSpaceSizeInBytes=*/reserve_space_data->size())); in DoRnnBackwardImpl()
1980 /*reserveSpace=*/reserve_space_data->opaque(), in DoRnnBackwardImpl()
1981 /*reserveSpaceSizeInBytes=*/reserve_space_data->size())); in DoRnnBackwardImpl()
1998 /*reserveSpace=*/reserve_space_data->opaque(), in DoRnnBackwardImpl()
1999 /*reserveSpaceSizeInBytes=*/reserve_space_data->size())); in DoRnnBackwardImpl()
2012 /*reserveSpace=*/reserve_space_data->opaque(), in DoRnnBackwardImpl()
2013 /*reserveSpaceSizeInBytes=*/reserve_space_data->size())); in DoRnnBackwardImpl()
2273 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackward() argument
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h176 DeviceMemory<uint8>* reserve_space_data,
201 DeviceMemory<uint8>* reserve_space_data,
226 DeviceMemory<uint8>* reserve_space_data,
292 DeviceMemory<uint8>* reserve_space_data,
303 DeviceMemory<uint8>* reserve_space_data,
776 DeviceMemory<uint8>* reserve_space_data,
Drocm_dnn.cc2218 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackwardImpl() argument
2284 reserve_space_data->opaque() /*reserveSpace*/, in DoRnnBackwardImpl()
2285 reserve_space_data->size() /*reserveSpaceSizeInBytes*/); in DoRnnBackwardImpl()
2303 reserve_space_data->opaque() /*reserveSpace*/, in DoRnnBackwardImpl()
2304 reserve_space_data->size() /*reserveSpaceSizeInBytes*/); in DoRnnBackwardImpl()
2706 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackward() argument
2733 reserve_space_data, workspace_allocator); in DoRnnBackward()
2757 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackward() argument
2784 reserve_space_data, workspace_allocator); in DoRnnBackward()
2809 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackward() argument
[all …]
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1072 DeviceMemory<uint8>* reserve_space_data, in DoBatchNormalizationBackward() argument
1088 DeviceMemory<uint8>* reserve_space_data, in DoBatchNormalizationBackward() argument
2310 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackward() argument
2338 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackward() argument
2366 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackward() argument
Dstream.h268 DeviceMemory<uint8> *reserve_space_data,
294 DeviceMemory<uint8> *reserve_space_data,
1858 DeviceMemory<uint8> *reserve_space_data,
1883 DeviceMemory<uint8> *reserve_space_data,
1908 DeviceMemory<uint8> *reserve_space_data,
Dstream.cc377 DeviceMemory<uint8> *reserve_space_data, in ThenBatchNormalizationBackward() argument
386 reserve_space_data, workspace_allocator)); in ThenBatchNormalizationBackward()
429 DeviceMemory<uint8> *reserve_space_data, in ThenBatchNormalizationBackward() argument
438 reserve_space_data, workspace_allocator)); in ThenBatchNormalizationBackward()
4661 DeviceMemory<uint8> *reserve_space_data, in ThenRnnBackward() argument
4672 params_backprop_data, reserve_space_data, workspace_allocator, in ThenRnnBackward()
4705 DeviceMemory<uint8> *reserve_space_data, in ThenRnnBackward() argument
4716 params_backprop_data, reserve_space_data, workspace_allocator, in ThenRnnBackward()
4750 DeviceMemory<uint8> *reserve_space_data, in ThenRnnBackward() argument
4761 params_backprop_data, reserve_space_data, workspace_allocator, in ThenRnnBackward()
/external/tensorflow/tensorflow/core/kernels/
Dfused_batch_norm_op.cc1109 DeviceMemory<uint8> reserve_space_data; in operator ()() local
1119 reserve_space_data = functor::CastDeviceMemory<uint8, U>( in operator ()()
1121 reserve_space_data_ptr = &reserve_space_data; in operator ()()