MCPcopy Create free account
hub / github.com/Oneflow-Inc/oneflow / fft_norm_from_string

Function fft_norm_from_string

oneflow/core/functional/impl/math_functor.cpp:3925–3936  ·  view source on GitHub ↗

Convert NumPy compatible normalization mode string to enum values In Numpy, "forward" translates to `by_n` for a forward transform and `none` for backward.

Source from the content-addressed store, hash-verified

3923// Convert NumPy compatible normalization mode string to enum values
3924// In Numpy, "forward" translates to `by_n` for a forward transform and `none` for backward.
3925static fft_norm_mode fft_norm_from_string(const Optional<std::string>& norm_op, bool forward) {
3926 std::string norm_str = norm_op.value_or("backward");
3927 if (norm_str == "backward") {
3928 return forward ? fft_norm_mode::none : fft_norm_mode::by_n;
3929 } else if (norm_str == "forward") {
3930 return forward ? fft_norm_mode::by_n : fft_norm_mode::none;
3931 } else if (norm_str == "ortho") {
3932 return fft_norm_mode::by_root_n;
3933 }
3934
3935 return fft_norm_mode::none;
3936}
3937
3938template<typename T>
3939static T fft_compute_fct(int64_t size, fft_norm_mode normalization) {

Callers 12

operator()Method · 0.85
operator()Method · 0.85
operator()Method · 0.85
operator()Method · 0.85
operator()Method · 0.85
operator()Method · 0.85
operator()Method · 0.85
operator()Method · 0.85
operator()Method · 0.85
operator()Method · 0.85
operator()Method · 0.85
operator()Method · 0.85

Calls 1

value_orMethod · 0.45

Tested by

no test coverage detected