Searched refs:normalized_data (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 517 const DeviceMemory<float>& normalized_data,
|
| D | cuda_dnn.cc | 4489 const DeviceMemory<float>& normalized_data, in DoNormalizeBackwardWithDimensions() argument
4513 &alpha, dims.handle(), normalized_data.opaque(), dims.handle(), in DoNormalizeBackwardWithDimensions()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 533 const DeviceMemory<float>& normalized_data,
|
| D | rocm_dnn.cc | 4385 const DeviceMemory<float>& normalized_data, in DoNormalizeBackwardWithDimensions() argument
4469 normalized_data.opaque(), dims.handle(), in DoNormalizeBackwardWithDimensions()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 1745 const DeviceMemory<float>& normalized_data, in DoNormalizeBackwardWithDimensions() argument
|
| D | stream.cc | 998 const DeviceMemory<float> &normalized_data, in ThenNormalizeBackwardWithDimensions() argument
1003 PARAM(normalized_data), PARAM(normalized_variable_gradient), in ThenNormalizeBackwardWithDimensions()
1008 this, normalize_descriptor, dimensions, raw_data, normalized_data, in ThenNormalizeBackwardWithDimensions()
|
| D | stream.h | 604 const DeviceMemory<float> &normalized_data,
|