Searched refs:NormalizeDescriptor (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | dnn.h | 1022 class NormalizeDescriptor {
1024 NormalizeDescriptor();
1026 NormalizeDescriptor& set_bias(float bias) { in set_bias()
1031 NormalizeDescriptor& set_range(int32_t range) { in set_range()
1036 NormalizeDescriptor& set_alpha(float alpha) { in set_alpha()
1041 NormalizeDescriptor& set_beta(float beta) { in set_beta()
1046 NormalizeDescriptor& set_wrap_around(bool wrap_around) { in set_wrap_around()
1051 NormalizeDescriptor& set_segment_size(int32_t segment_size) { in set_segment_size()
1056 void CloneFrom(const NormalizeDescriptor& other);
1647 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor, in DoNormalizeWithDimensions()
[all …]
|
| H A D | dnn.cc | 835 NormalizeDescriptor::NormalizeDescriptor() in NormalizeDescriptor() function in stream_executor::dnn::NormalizeDescriptor
843 void NormalizeDescriptor::CloneFrom(const NormalizeDescriptor& other) { in CloneFrom()
852 std::string NormalizeDescriptor::ToString() const { in ToString()
859 std::string NormalizeDescriptor::ToShortString() const { in ToShortString()
|
| H A D | stream.h | 504 const dnn::NormalizeDescriptor &normalize_descriptor,
509 const dnn::NormalizeDescriptor &normalize_descriptor,
|
| H A D | stream.cc | 55 std::string ToVlogString(const dnn::NormalizeDescriptor &descriptor) { in ToVlogString()
610 const dnn::NormalizeDescriptor &normalize_descriptor, in ThenNormalizeWithDimensions()
626 const dnn::NormalizeDescriptor &normalize_descriptor, in ThenNormalizeBackwardWithDimensions()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | lrn_op.cc | 210 se::dnn::NormalizeDescriptor normalize_desc; in launch()
263 perftools::gputools::dnn::NormalizeDescriptor normalize_desc; in launch()
502 se::dnn::NormalizeDescriptor normalize_desc; in launch()
577 perftools::gputools::dnn::NormalizeDescriptor normalize_desc; in launch()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_dnn.h | 457 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor,
463 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor,
|
| H A D | cuda_dnn.cc | 978 const dnn::NormalizeDescriptor& normalize_descriptor) in CudnnNormalizeDescriptor()
6362 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor, in DoNormalizeWithDimensions()
6397 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor, in DoNormalizeBackwardWithDimensions()
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_dnn.h | 461 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor,
467 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor,
|
| H A D | rocm_dnn.cc | 71 using dnn::NormalizeDescriptor;
937 ScopedNormalizeDescriptor(const NormalizeDescriptor& normalize_descriptor) in ScopedNormalizeDescriptor()
4286 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor, in DoNormalizeWithDimensions()
4322 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor, in DoNormalizeBackwardWithDimensions()
|