Searched refs:raw_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 | 468 DeviceMemory<float>* raw_variable_gradient,
|
H A D | cuda_dnn.cc | 6401 DeviceMemory<float>* raw_variable_gradient, in DoNormalizeBackwardWithDimensions() argument 6425 &beta, dims.handle(), raw_variable_gradient->opaque())); in DoNormalizeBackwardWithDimensions()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
H A D | stream.cc | 630 DeviceMemory<float> *raw_variable_gradient, in ThenNormalizeBackwardWithDimensions() argument 634 PARAM(raw_variable_gradient), PARAM(workspace_allocator)); in ThenNormalizeBackwardWithDimensions() 639 normalized_variable_gradient, raw_variable_gradient, in ThenNormalizeBackwardWithDimensions()
|
H A D | dnn.h | 1671 DeviceMemory<float>* raw_variable_gradient, in DoNormalizeBackwardWithDimensions() argument
|
H A D | stream.h | 514 DeviceMemory<float> *raw_variable_gradient,
|
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
H A D | rocm_dnn.h | 472 DeviceMemory<float>* raw_variable_gradient,
|
H A D | rocm_dnn.cc | 4326 DeviceMemory<float>* raw_variable_gradient, in DoNormalizeBackwardWithDimensions() argument 4410 &beta, dims.handle(), raw_variable_gradient->opaque(), in DoNormalizeBackwardWithDimensions()
|