Searched refs:NormalizeDescriptor (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.cc | 583 NormalizeDescriptor::NormalizeDescriptor() in NormalizeDescriptor() function in stream_executor::dnn::NormalizeDescriptor
591 void NormalizeDescriptor::CloneFrom(const NormalizeDescriptor& other) { in CloneFrom()
600 std::string NormalizeDescriptor::ToString() const { in ToString()
607 std::string NormalizeDescriptor::ToShortString() const { in ToShortString()
|
| D | dnn.h | 871 class NormalizeDescriptor {
873 NormalizeDescriptor();
875 NormalizeDescriptor& set_bias(float bias) { in set_bias()
880 NormalizeDescriptor& set_range(int32 range) { in set_range()
885 NormalizeDescriptor& set_alpha(float alpha) { in set_alpha()
890 NormalizeDescriptor& set_beta(float beta) { in set_beta()
895 NormalizeDescriptor& set_wrap_around(bool wrap_around) { in set_wrap_around()
900 NormalizeDescriptor& set_segment_size(int32 segment_size) { in set_segment_size()
905 void CloneFrom(const NormalizeDescriptor& other);
1723 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor, in DoNormalizeWithDimensions()
[all …]
|
| D | stream.h | 596 const dnn::NormalizeDescriptor &normalize_descriptor,
601 const dnn::NormalizeDescriptor &normalize_descriptor,
|
| D | stream.cc | 53 std::string ToVlogString(const dnn::NormalizeDescriptor &descriptor) { in ToVlogString()
980 const dnn::NormalizeDescriptor &normalize_descriptor, in ThenNormalizeWithDimensions()
996 const dnn::NormalizeDescriptor &normalize_descriptor, in ThenNormalizeBackwardWithDimensions()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| 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()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 508 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor,
514 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor,
|
| D | cuda_dnn.cc | 821 const dnn::NormalizeDescriptor& normalize_descriptor) in CudnnNormalizeDescriptor()
4452 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor, in DoNormalizeWithDimensions()
4487 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor, in DoNormalizeBackwardWithDimensions()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 524 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor,
530 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor,
|
| D | rocm_dnn.cc | 67 using dnn::NormalizeDescriptor;
883 ScopedNormalizeDescriptor(const NormalizeDescriptor& normalize_descriptor) in ScopedNormalizeDescriptor()
4347 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor, in DoNormalizeWithDimensions()
4383 Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor, in DoNormalizeBackwardWithDimensions()
|