Searched refs:normalized_variable_gradient (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_dnn.h | 467 const DeviceMemory<float>& normalized_variable_gradient,
|
| H A D | cuda_dnn.cc | 6400 const DeviceMemory<float>& normalized_variable_gradient, in DoNormalizeBackwardWithDimensions() argument
6424 normalized_variable_gradient.opaque(), dims.handle(), raw_data.opaque(), in DoNormalizeBackwardWithDimensions()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | stream.cc | 629 const DeviceMemory<float> &normalized_variable_gradient, in ThenNormalizeBackwardWithDimensions() argument
633 PARAM(normalized_data), PARAM(normalized_variable_gradient), in ThenNormalizeBackwardWithDimensions()
639 normalized_variable_gradient, raw_variable_gradient, in ThenNormalizeBackwardWithDimensions()
|
| H A D | dnn.h | 1670 const DeviceMemory<float>& normalized_variable_gradient, in DoNormalizeBackwardWithDimensions() argument
|
| H A D | stream.h | 513 const DeviceMemory<float> &normalized_variable_gradient,
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_dnn.h | 471 const DeviceMemory<float>& normalized_variable_gradient,
|
| H A D | rocm_dnn.cc | 4325 const DeviceMemory<float>& normalized_variable_gradient, in DoNormalizeBackwardWithDimensions() argument
4409 normalized_variable_gradient.opaque(), dims.handle(), raw_data.opaque(), in DoNormalizeBackwardWithDimensions()
|