1 /** 2 * Copyright 2020 Huawei Technologies Co., Ltd 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 #include "ops/scale.h" 18 19 namespace mindspore { 20 namespace ops { Init(const int64_t axis)21void Scale::Init(const int64_t axis) { set_axis(axis); } 22 set_axis(const int64_t axis)23void Scale::set_axis(const int64_t axis) { (void)this->AddAttr(kAxis, MakeValue(axis)); } get_axis() const24int64_t Scale::get_axis() const { return GetValue<int64_t>(GetAttr(kAxis)); } 25 26 REGISTER_PRIMITIVE_C(kNameScale, Scale); 27 } // namespace ops 28 } // namespace mindspore 29