Home
last modified time | relevance | path

Searched defs:reserve_space_data (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1072 DeviceMemory<uint8>* reserve_space_data, in DoBatchNormalizationBackward()
1088 DeviceMemory<uint8>* reserve_space_data, in DoBatchNormalizationBackward()
2310 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackward()
2338 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackward()
2366 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackward()
Dstream.cc377 DeviceMemory<uint8> *reserve_space_data, in ThenBatchNormalizationBackward()
429 DeviceMemory<uint8> *reserve_space_data, in ThenBatchNormalizationBackward()
4661 DeviceMemory<uint8> *reserve_space_data, in ThenRnnBackward()
4705 DeviceMemory<uint8> *reserve_space_data, in ThenRnnBackward()
4750 DeviceMemory<uint8> *reserve_space_data, in ThenRnnBackward()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1907 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackwardImpl()
2273 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackward()
2324 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackward()
2376 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackward()
3650 DeviceMemory<uint8>* reserve_space_data, in DoBatchNormalizationBackward()
3668 DeviceMemory<uint8>* reserve_space_data, in DoBatchNormalizationBackward()
3686 DeviceMemory<U>* offset_backprop, DeviceMemory<uint8>* reserve_space_data, in DoBatchNormalizationBackwardImpl()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2218 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackwardImpl()
2706 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackward()
2757 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackward()
2809 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackward()
3618 DeviceMemory<uint8>* reserve_space_data, in DoBatchNormalizationBackward()
3634 DeviceMemory<uint8>* reserve_space_data, in DoBatchNormalizationBackward()
/external/tensorflow/tensorflow/core/kernels/
Dfused_batch_norm_op.cc1109 DeviceMemory<uint8> reserve_space_data; in operator ()() local