Home
last modified time | relevance | path

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 Dcuda_dnn.h468 DeviceMemory<float>* raw_variable_gradient,
H A Dcuda_dnn.cc6401 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 Dstream.cc630 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 Ddnn.h1671 DeviceMemory<float>* raw_variable_gradient, in DoNormalizeBackwardWithDimensions() argument
H A Dstream.h514 DeviceMemory<float> *raw_variable_gradient,
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.h472 DeviceMemory<float>* raw_variable_gradient,
H A Drocm_dnn.cc4326 DeviceMemory<float>* raw_variable_gradient, in DoNormalizeBackwardWithDimensions() argument
4410 &beta, dims.handle(), raw_variable_gradient->opaque(), in DoNormalizeBackwardWithDimensions()