|
|
|
@@ -30,6 +30,8 @@ |
|
|
|
|
|
|
|
namespace mindspore { |
|
|
|
// op name. Op which not exists in operator/ops.h, so define it's name here |
|
|
|
constexpr auto kUniqueOpName = "Unique"; |
|
|
|
constexpr auto kComputeAccidentalHitsOpName = "ComputeAccidentalHits"; |
|
|
|
constexpr auto kFour2FiveOpName = "Four2Five"; |
|
|
|
constexpr auto kFive2FourOpName = "Five2Four"; |
|
|
|
constexpr auto kConv2DOpName = "Conv2D"; |
|
|
|
@@ -479,6 +481,8 @@ const std::set<std::string> kHWSpecialFormatSet = { |
|
|
|
|
|
|
|
const std::set<TypeId> kFloatDataTypeSet = {kNumberTypeFloat16, kNumberTypeFloat32}; |
|
|
|
|
|
|
|
const std::set<std::string> kComputeDepend = {kUniqueOpName, kComputeAccidentalHitsOpName}; |
|
|
|
|
|
|
|
static inline void ChangeFileMode(const std::string &file_name, mode_t mode) { |
|
|
|
try { |
|
|
|
if (chmod(file_name.c_str(), mode) != 0) { |
|
|
|
|