Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Ddnn.h1226 DeviceMemory<uint8>* reserve_space_data, in DoBatchNormalizationBackward()
1245 DeviceMemory<uint8>* reserve_space_data, in DoBatchNormalizationBackward()
2242 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackward()
2271 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackward()
2300 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackward()
Dstream.cc364 DeviceMemory<uint8> *reserve_space_data, in ThenBatchNormalizationBackward()
419 DeviceMemory<uint8> *reserve_space_data, in ThenBatchNormalizationBackward()
2160 DeviceMemory<uint8> *reserve_space_data, in ThenRnnBackward()
2205 DeviceMemory<uint8> *reserve_space_data, in ThenRnnBackward()
2251 DeviceMemory<uint8> *reserve_space_data, in ThenRnnBackward()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_dnn.cc2164 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackwardImpl()
2623 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackward()
2675 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackward()
2728 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackward()
5614 DeviceMemory<uint8>* reserve_space_data, in DoBatchNormalizationBackward()
5635 DeviceMemory<uint8>* reserve_space_data, in DoBatchNormalizationBackward()
5657 DeviceMemory<uint8>* reserve_space_data, in DoBatchNormalizationBackwardImpl()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2272 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackwardImpl()
2764 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackward()
2816 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackward()
2869 DeviceMemory<uint8>* reserve_space_data, in DoRnnBackward()
3736 DeviceMemory<uint8>* reserve_space_data, in DoBatchNormalizationBackward()
3754 DeviceMemory<uint8>* reserve_space_data, in DoBatchNormalizationBackward()
/external/tensorflow/tensorflow/core/kernels/
Dfused_batch_norm_op.cc1132 DeviceMemory<uint8> reserve_space_data; in operator ()() local