Searched defs:backward_output_data (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 3075 DeviceMemory<T> backward_output_data, in MaybeTransformLayout()
3141 DeviceMemory<T> backward_output_data, in DoConvolveBackwardDataImpl()
3351 DeviceMemory<float> backward_output_data, in DoConvolveBackwardData()
3369 DeviceMemory<Eigen::half> backward_output_data, in DoConvolveBackwardData()
3388 DeviceMemory<T> backward_output_data, in DoConvolveBackwardFilterImpl()
3599 DeviceMemory<float> backward_output_data, in DoConvolveBackwardFilter()
3617 DeviceMemory<Eigen::half> backward_output_data, in DoConvolveBackwardFilter()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 869 DeviceMemory<float> backward_output_data, in ThenConvolveBackwardDataWithScratch()
897 DeviceMemory<float> backward_output_data, in ThenConvolveBackwardDataWithAlgorithm()
930 DeviceMemory<Eigen::half> backward_output_data, in ThenConvolveBackwardDataWithAlgorithm()
963 DeviceMemory<Eigen::half> backward_output_data, in ThenConvolveBackwardDataWithScratch()
991 DeviceMemory<float> backward_output_data, in ThenConvolveBackwardData()
1005 DeviceMemory<float> backward_output_data, in ThenConvolveBackwardFilterWithScratch()
1033 DeviceMemory<float> backward_output_data, in ThenConvolveBackwardFilterWithAlgorithm()
1066 DeviceMemory<Eigen::half> backward_output_data, in ThenConvolveBackwardFilterWithScratch()
1094 DeviceMemory<Eigen::half> backward_output_data, in ThenConvolveBackwardFilterWithAlgorithm()
1127 DeviceMemory<float> backward_output_data, in ThenConvolveBackwardFilter()
|