Searched refs:raw_variable_gradient (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 500 DeviceMemory<float>* raw_variable_gradient,
|
| D | cuda_dnn.cc | 3958 DeviceMemory<float>* raw_variable_gradient, in DoNormalizeBackwardWithDimensions() argument
3982 &beta, dims.handle(), raw_variable_gradient->opaque())); in DoNormalizeBackwardWithDimensions()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 483 DeviceMemory<float>* raw_variable_gradient,
|
| D | rocm_dnn.cc | 3790 DeviceMemory<float>* raw_variable_gradient, in DoNormalizeBackwardWithDimensions() argument
3874 &beta, dims.handle(), raw_variable_gradient->opaque(), in DoNormalizeBackwardWithDimensions()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 1681 DeviceMemory<float>* raw_variable_gradient,
|
| D | stream.cc | 1419 DeviceMemory<float> *raw_variable_gradient, in ThenNormalizeBackwardWithDimensions() argument
1423 PARAM(raw_variable_gradient), PARAM(workspace_allocator)); in ThenNormalizeBackwardWithDimensions()
1429 normalized_variable_gradient, raw_variable_gradient, in ThenNormalizeBackwardWithDimensions()
|
| D | stream.h | 583 DeviceMemory<float> *raw_variable_gradient,
|