Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.cc937 ScopedNormalizeDescriptor(const NormalizeDescriptor& normalize_descriptor) in ScopedNormalizeDescriptor() argument
949 unsigned lrn_N = 2 * normalize_descriptor.range() + 1; in ScopedNormalizeDescriptor()
961 double lrn_alpha = lrn_N * normalize_descriptor.alpha(); in ScopedNormalizeDescriptor()
963 double lrn_beta = normalize_descriptor.beta(); in ScopedNormalizeDescriptor()
964 double lrn_k = normalize_descriptor.bias(); in ScopedNormalizeDescriptor()
4286 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor, in DoNormalizeWithDimensions() argument
4290 if (normalize_descriptor.wrap_around()) { in DoNormalizeWithDimensions()
4294 if (normalize_descriptor.segment_size()) { in DoNormalizeWithDimensions()
4303 ScopedNormalizeDescriptor normalize{normalize_descriptor}; in DoNormalizeWithDimensions()
4322 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor, in DoNormalizeBackwardWithDimensions() argument
[all …]
H A Drocm_dnn.h461 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor,
467 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor,
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.cc978 const dnn::NormalizeDescriptor& normalize_descriptor) in CudnnNormalizeDescriptor() argument
984 unsigned lrnN = 2 * normalize_descriptor.range() + 1; in CudnnNormalizeDescriptor()
996 double lrnAlpha = lrnN * normalize_descriptor.alpha(); in CudnnNormalizeDescriptor()
998 double lrnBeta = normalize_descriptor.beta(); in CudnnNormalizeDescriptor()
999 double lrnK = normalize_descriptor.bias(); in CudnnNormalizeDescriptor()
6362 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor, in DoNormalizeWithDimensions() argument
6366 if (normalize_descriptor.wrap_around()) { in DoNormalizeWithDimensions()
6370 if (normalize_descriptor.segment_size()) { in DoNormalizeWithDimensions()
6376 CudnnNormalizeDescriptor normalize(normalize_descriptor); in DoNormalizeWithDimensions()
6397 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor, in DoNormalizeBackwardWithDimensions() argument
[all …]
H A Dcuda_dnn.h457 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor,
463 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor,
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Dstream.cc610 const dnn::NormalizeDescriptor &normalize_descriptor, in ThenNormalizeWithDimensions() argument
613 VLOG_CALL(PARAM(normalize_descriptor), PARAM(dimensions), PARAM(input_data), in ThenNormalizeWithDimensions()
618 this, normalize_descriptor, dimensions, input_data, output_data)); in ThenNormalizeWithDimensions()
626 const dnn::NormalizeDescriptor &normalize_descriptor, in ThenNormalizeBackwardWithDimensions() argument
632 VLOG_CALL(PARAM(normalize_descriptor), PARAM(dimensions), PARAM(raw_data), in ThenNormalizeBackwardWithDimensions()
638 this, normalize_descriptor, dimensions, raw_data, normalized_data, in ThenNormalizeBackwardWithDimensions()
H A Ddnn.h1647 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor, in DoNormalizeWithDimensions() argument
1666 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor, in DoNormalizeBackwardWithDimensions() argument
H A Dstream.h504 const dnn::NormalizeDescriptor &normalize_descriptor,
509 const dnn::NormalizeDescriptor &normalize_descriptor,