Browse Source

!14972 fix codedex

From: @he-botao
Reviewed-by: @oacjiewen,@wuxuejian
Signed-off-by: @wuxuejian
pull/14972/MERGE
mindspore-ci-bot Gitee 4 years ago
parent
commit
e8c59b98c9
4 changed files with 2 additions and 3 deletions
  1. +2
    -0
      mindspore/ccsrc/backend/kernel_compiler/cpu/cpu_kernel.h
  2. +0
    -1
      mindspore/ccsrc/backend/kernel_compiler/cpu/dropout_cpu_kernel.h
  3. +0
    -1
      mindspore/ccsrc/backend/kernel_compiler/cpu/isfinite_cpu_kernel.cc
  4. +0
    -1
      mindspore/ccsrc/backend/kernel_compiler/cpu/isfinite_cpu_kernel.h

+ 2
- 0
mindspore/ccsrc/backend/kernel_compiler/cpu/cpu_kernel.h View File

@@ -150,6 +150,7 @@ class BroadcastIterator {
public:
BroadcastIterator(std::vector<size_t> input_shape_a, std::vector<size_t> input_shape_b,
std::vector<size_t> output_shape);
virtual ~BroadcastIterator() = default;
inline size_t GetInputPosA() const { return input_pos_[0]; }
inline size_t GetInputPosB() const { return input_pos_[1]; }
void SetPos(size_t pos);
@@ -174,6 +175,7 @@ class BroadcastIterator {
class TransposeIterator {
public:
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_; }
void SetPos(size_t pos);
void GenNextPos();


+ 0
- 1
mindspore/ccsrc/backend/kernel_compiler/cpu/dropout_cpu_kernel.h View File

@@ -54,7 +54,6 @@ MS_REG_CPU_KERNEL(
Dropout,
KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumberTypeFloat32).AddOutputAttr(kNumberTypeFloat32),
DropoutCPUKernel);

} // namespace kernel
} // namespace mindspore
#endif // MINDSPORE_CCSRC_BACKEND_KERNEL_COMPILER_CPU_DROPOUT_CPU_KERNEL_H_

+ 0
- 1
mindspore/ccsrc/backend/kernel_compiler/cpu/isfinite_cpu_kernel.cc View File

@@ -90,6 +90,5 @@ void IsFiniteCPUKernel::LaunchKernelOther(const std::vector<AddressPtr> &inputs,
output[i] = true;
}
}

} // namespace kernel
} // namespace mindspore

+ 0
- 1
mindspore/ccsrc/backend/kernel_compiler/cpu/isfinite_cpu_kernel.h View File

@@ -86,7 +86,6 @@ MS_REG_CPU_KERNEL(IsFinite, KernelAttr().AddInputAttr(kNumberTypeUInt32).AddOutp

MS_REG_CPU_KERNEL(IsFinite, KernelAttr().AddInputAttr(kNumberTypeUInt64).AddOutputAttr(kNumberTypeBool),
IsFiniteCPUKernel);

} // namespace kernel
} // namespace mindspore



Loading…
Cancel
Save