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.cc760 const dnn::NormalizeDescriptor& normalize_descriptor) in CudnnNormalizeDescriptor() argument
766 unsigned lrnN = 2 * normalize_descriptor.range() + 1; in CudnnNormalizeDescriptor()
778 double lrnAlpha = lrnN * normalize_descriptor.alpha(); in CudnnNormalizeDescriptor()
780 double lrnBeta = normalize_descriptor.beta(); in CudnnNormalizeDescriptor()
781 double lrnK = normalize_descriptor.bias(); in CudnnNormalizeDescriptor()
3919 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor, in DoNormalizeWithDimensions() argument
3923 if (normalize_descriptor.wrap_around()) { in DoNormalizeWithDimensions()
3927 if (normalize_descriptor.segment_size()) { in DoNormalizeWithDimensions()
3933 CudnnNormalizeDescriptor normalize(normalize_descriptor); in DoNormalizeWithDimensions()
3954 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor, in DoNormalizeBackwardWithDimensions() argument
[all …]
Dcuda_dnn.h489 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor,
495 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor,
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc766 ScopedNormalizeDescriptor(const NormalizeDescriptor& normalize_descriptor) in ScopedNormalizeDescriptor() argument
778 unsigned lrn_N = 2 * normalize_descriptor.range() + 1; in ScopedNormalizeDescriptor()
790 double lrn_alpha = lrn_N * normalize_descriptor.alpha(); in ScopedNormalizeDescriptor()
792 double lrn_beta = normalize_descriptor.beta(); in ScopedNormalizeDescriptor()
793 double lrn_k = normalize_descriptor.bias(); in ScopedNormalizeDescriptor()
3750 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor, in DoNormalizeWithDimensions() argument
3754 if (normalize_descriptor.wrap_around()) { in DoNormalizeWithDimensions()
3758 if (normalize_descriptor.segment_size()) { in DoNormalizeWithDimensions()
3767 ScopedNormalizeDescriptor normalize{normalize_descriptor}; in DoNormalizeWithDimensions()
3786 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor, in DoNormalizeBackwardWithDimensions() argument
[all …]
Drocm_dnn.h472 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor,
478 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor,
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1657 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor,
1676 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor,
Dstream.cc1397 const dnn::NormalizeDescriptor &normalize_descriptor, in ThenNormalizeWithDimensions() argument
1400 VLOG_CALL(PARAM(normalize_descriptor), PARAM(dimensions), PARAM(input_data), in ThenNormalizeWithDimensions()
1406 this, normalize_descriptor, dimensions, input_data, output_data)); in ThenNormalizeWithDimensions()
1415 const dnn::NormalizeDescriptor &normalize_descriptor, in ThenNormalizeBackwardWithDimensions() argument
1421 VLOG_CALL(PARAM(normalize_descriptor), PARAM(dimensions), PARAM(raw_data), in ThenNormalizeBackwardWithDimensions()
1428 this, normalize_descriptor, dimensions, raw_data, normalized_data, in ThenNormalizeBackwardWithDimensions()
Dstream.h573 const dnn::NormalizeDescriptor &normalize_descriptor,
578 const dnn::NormalizeDescriptor &normalize_descriptor,