text: ﻿// Copyright (c) 2018-2019 Intel Corporation
  //
  // Licensed under the Apache License, Version 2.0 (the "License");
  // you may not use this file except in compliance with the License.
  // You may obtain a copy of the License at
  //
  //      http://www.apache.org/licenses/LICENSE-2.0
  //
  // Unless required by applicable law or agreed to in writing, software
  // distributed under the License is distributed on an "AS IS" BASIS,
  // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  // See the License for the specific language governing permissions and
  // limitations under the License.
  
  
  #include "mvn_kernel_ref.h"
  #include <string>
  
  namespace kernel_selector {
  ParamsKey MVNKernelRef::GetSupportedKey() const {
      ParamsKey k;
      k.EnableInputDataType(Datatype::F16);
      k.EnableInputDataType(Datatype::F32);
      k.EnableOutputDataType(Datatype::F16);
      k.EnableOutputDataType(Datatype::F32);
      k.EnableInputLayout(DataLayout::bfyx);
      k.EnableInputLayout(DataLayout::yxfb);
      k.EnableInputLayout(DataLayout::byxf);
      k.EnableOutputLayout(DataLayout::bfyx);
      k.EnableOutputLayout(DataLayout::yxfb);
      k.EnableOutputLayout(DataLayout::byxf);
      k.EnableInputLayout(DataLayout::bfzyx);
      k.EnableOutputLayout(DataLayout::bfzyx);
      k.EnableTensorOffset();
      k.EnableTensorPitches();
      k.EnableBatching();
      k.EnableMVNMode(MVNMode::ACROSS_CHANNELS);
      k.EnableMVNMode(MVNMode::WITHIN_CHANNELS);
      k.EnableMVNNormalizeVariance();
      return k;
  }
  
  std::string MVNKernelRef::GetKernelName(const mvn_params& params) const {
      if (params.mvnMode == MVNMode::ACROSS_CHANNELS)
          return kernelName + "_accross_channels";
      else
          return kernelName + "_within_channels";
  }
  
  KernelsData MVNKernelRef::GetKernelsData(const Params& params, const optional_params& optParams) const {
      return GetCommonKernelsData(params, optParams, FORCE_PRIORITY_9);
  }
  }  // namespace kernel_selector