From: @he-botao Reviewed-by: @oacjiewen,@wuxuejian Signed-off-by: @wuxuejianpull/14972/MERGE
| @@ -150,6 +150,7 @@ class BroadcastIterator { | |||||
| public: | public: | ||||
| BroadcastIterator(std::vector<size_t> input_shape_a, std::vector<size_t> input_shape_b, | BroadcastIterator(std::vector<size_t> input_shape_a, std::vector<size_t> input_shape_b, | ||||
| std::vector<size_t> output_shape); | std::vector<size_t> output_shape); | ||||
| virtual ~BroadcastIterator() = default; | |||||
| inline size_t GetInputPosA() const { return input_pos_[0]; } | inline size_t GetInputPosA() const { return input_pos_[0]; } | ||||
| inline size_t GetInputPosB() const { return input_pos_[1]; } | inline size_t GetInputPosB() const { return input_pos_[1]; } | ||||
| void SetPos(size_t pos); | void SetPos(size_t pos); | ||||
| @@ -174,6 +175,7 @@ class BroadcastIterator { | |||||
| class TransposeIterator { | class TransposeIterator { | ||||
| public: | public: | ||||
| TransposeIterator(std::vector<size_t> output_shape, std::vector<size_t> axes, const std::vector<size_t> &input_shape); | TransposeIterator(std::vector<size_t> output_shape, std::vector<size_t> axes, const std::vector<size_t> &input_shape); | ||||
| virtual ~TransposeIterator() = default; | |||||
| inline size_t GetPos() const { return pos_; } | inline size_t GetPos() const { return pos_; } | ||||
| void SetPos(size_t pos); | void SetPos(size_t pos); | ||||
| void GenNextPos(); | void GenNextPos(); | ||||
| @@ -54,7 +54,6 @@ MS_REG_CPU_KERNEL( | |||||
| Dropout, | Dropout, | ||||
| KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumberTypeFloat32).AddOutputAttr(kNumberTypeFloat32), | KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumberTypeFloat32).AddOutputAttr(kNumberTypeFloat32), | ||||
| DropoutCPUKernel); | DropoutCPUKernel); | ||||
| } // namespace kernel | } // namespace kernel | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| #endif // MINDSPORE_CCSRC_BACKEND_KERNEL_COMPILER_CPU_DROPOUT_CPU_KERNEL_H_ | #endif // MINDSPORE_CCSRC_BACKEND_KERNEL_COMPILER_CPU_DROPOUT_CPU_KERNEL_H_ | ||||
| @@ -90,6 +90,5 @@ void IsFiniteCPUKernel::LaunchKernelOther(const std::vector<AddressPtr> &inputs, | |||||
| output[i] = true; | output[i] = true; | ||||
| } | } | ||||
| } | } | ||||
| } // namespace kernel | } // namespace kernel | ||||
| } // namespace mindspore | } // namespace mindspore | ||||
| @@ -86,7 +86,6 @@ MS_REG_CPU_KERNEL(IsFinite, KernelAttr().AddInputAttr(kNumberTypeUInt32).AddOutp | |||||
| MS_REG_CPU_KERNEL(IsFinite, KernelAttr().AddInputAttr(kNumberTypeUInt64).AddOutputAttr(kNumberTypeBool), | MS_REG_CPU_KERNEL(IsFinite, KernelAttr().AddInputAttr(kNumberTypeUInt64).AddOutputAttr(kNumberTypeBool), | ||||
| IsFiniteCPUKernel); | IsFiniteCPUKernel); | ||||
| } // namespace kernel | } // namespace kernel | ||||
| } // namespace mindspore | } // namespace mindspore | ||||