Searched refs:NormalizeDescriptor (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.cc | 554 NormalizeDescriptor::NormalizeDescriptor() in NormalizeDescriptor() function in stream_executor::dnn::NormalizeDescriptor
562 void NormalizeDescriptor::CloneFrom(const NormalizeDescriptor& other) { in CloneFrom()
571 string NormalizeDescriptor::ToString() const { in ToString()
578 string NormalizeDescriptor::ToShortString() const { in ToShortString()
|
| D | dnn.h | 849 class NormalizeDescriptor {
851 NormalizeDescriptor();
853 NormalizeDescriptor& set_bias(float bias) {
858 NormalizeDescriptor& set_range(int32 range) {
863 NormalizeDescriptor& set_alpha(float alpha) {
868 NormalizeDescriptor& set_beta(float beta) {
873 NormalizeDescriptor& set_wrap_around(bool wrap_around) {
878 NormalizeDescriptor& set_segment_size(int32 segment_size) {
883 void CloneFrom(const NormalizeDescriptor& other);
1657 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor,
[all …]
|
| D | stream.h | 573 const dnn::NormalizeDescriptor &normalize_descriptor,
578 const dnn::NormalizeDescriptor &normalize_descriptor,
|
| D | stream.cc | 54 string ToVlogString(const dnn::NormalizeDescriptor &descriptor) { in ToVlogString()
1397 const dnn::NormalizeDescriptor &normalize_descriptor, in ThenNormalizeWithDimensions()
1415 const dnn::NormalizeDescriptor &normalize_descriptor, in ThenNormalizeBackwardWithDimensions()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | lrn_op.cc | 201 se::dnn::NormalizeDescriptor normalize_desc; in launch()
418 se::dnn::NormalizeDescriptor normalize_desc; in launch()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 489 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor,
495 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor,
|
| D | cuda_dnn.cc | 760 const dnn::NormalizeDescriptor& normalize_descriptor) in CudnnNormalizeDescriptor()
3919 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor, in DoNormalizeWithDimensions()
3954 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor, in DoNormalizeBackwardWithDimensions()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 472 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor,
478 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor,
|
| D | rocm_dnn.cc | 63 using dnn::NormalizeDescriptor;
766 ScopedNormalizeDescriptor(const NormalizeDescriptor& normalize_descriptor) in ScopedNormalizeDescriptor()
3750 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor, in DoNormalizeWithDimensions()
3786 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor, in DoNormalizeBackwardWithDimensions()
|