Home
last modified time | relevance | path

Searched refs:normalize_descriptor (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc821 const dnn::NormalizeDescriptor& normalize_descriptor) in CudnnNormalizeDescriptor() argument
827 unsigned lrnN = 2 * normalize_descriptor.range() + 1; in CudnnNormalizeDescriptor()
839 double lrnAlpha = lrnN * normalize_descriptor.alpha(); in CudnnNormalizeDescriptor()
841 double lrnBeta = normalize_descriptor.beta(); in CudnnNormalizeDescriptor()
842 double lrnK = normalize_descriptor.bias(); in CudnnNormalizeDescriptor()
4452 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor, in DoNormalizeWithDimensions() argument
4456 if (normalize_descriptor.wrap_around()) { in DoNormalizeWithDimensions()
4460 if (normalize_descriptor.segment_size()) { in DoNormalizeWithDimensions()
4466 CudnnNormalizeDescriptor normalize(normalize_descriptor); in DoNormalizeWithDimensions()
4487 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor, in DoNormalizeBackwardWithDimensions() argument
[all …]
Dcuda_dnn.h508 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor,
514 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor,
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc883 ScopedNormalizeDescriptor(const NormalizeDescriptor& normalize_descriptor) in ScopedNormalizeDescriptor() argument
895 unsigned lrn_N = 2 * normalize_descriptor.range() + 1; in ScopedNormalizeDescriptor()
907 double lrn_alpha = lrn_N * normalize_descriptor.alpha(); in ScopedNormalizeDescriptor()
909 double lrn_beta = normalize_descriptor.beta(); in ScopedNormalizeDescriptor()
910 double lrn_k = normalize_descriptor.bias(); in ScopedNormalizeDescriptor()
4347 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor, in DoNormalizeWithDimensions() argument
4351 if (normalize_descriptor.wrap_around()) { in DoNormalizeWithDimensions()
4355 if (normalize_descriptor.segment_size()) { in DoNormalizeWithDimensions()
4364 ScopedNormalizeDescriptor normalize{normalize_descriptor}; in DoNormalizeWithDimensions()
4383 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor, in DoNormalizeBackwardWithDimensions() argument
[all …]
Drocm_dnn.h524 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor,
530 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor,
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1723 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor, in DoNormalizeWithDimensions() argument
1742 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor, in DoNormalizeBackwardWithDimensions() argument
Dstream.cc980 const dnn::NormalizeDescriptor &normalize_descriptor, in ThenNormalizeWithDimensions() argument
983 VLOG_CALL(PARAM(normalize_descriptor), PARAM(dimensions), PARAM(input_data), in ThenNormalizeWithDimensions()
988 this, normalize_descriptor, dimensions, input_data, output_data)); in ThenNormalizeWithDimensions()
996 const dnn::NormalizeDescriptor &normalize_descriptor, in ThenNormalizeBackwardWithDimensions() argument
1002 VLOG_CALL(PARAM(normalize_descriptor), PARAM(dimensions), PARAM(raw_data), in ThenNormalizeBackwardWithDimensions()
1008 this, normalize_descriptor, dimensions, raw_data, normalized_data, in ThenNormalizeBackwardWithDimensions()
Dstream.h596 const dnn::NormalizeDescriptor &normalize_descriptor,
601 const dnn::NormalizeDescriptor &normalize_descriptor,