Home
last modified time | relevance | path

Searched refs:kDeviceTypeNone (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/lite/tools/optimizer/fisson/
Dmulti_conv_split_pass.cc65 …evice_type = (device_type_attr != nullptr) ? GetValue<int32_t>(device_type_attr) : kDeviceTypeNone; in Process()
66 if (device_type != kDeviceTypeNone) { in Process()
/third_party/mindspore/mindspore/lite/tools/optimizer/parallel/
Dspliter.cc76 …tr(ops::kDeviceType) != nullptr ? GetValue<int>(prim->GetAttr(ops::kDeviceType)) : kDeviceTypeNone; in RecordGraphInfo()
78 if (device_type != kDeviceTypeNone) { in RecordGraphInfo()
Dsplit_strategy.h42 constexpr auto kDeviceTypeNone = -1; variable
/third_party/mindspore/mindspore/lite/tools/optimizer/common/
Dgllo_utils.cc39 constexpr auto kDeviceTypeNone = -1; variable
695 …ps::kDeviceType) != nullptr ? GetValue<int32_t>(prim->GetAttr(ops::kDeviceType)) : kDeviceTypeNone; in IsParallelSplitConvNode()
696 if (device_type != kDeviceTypeNone) { in IsParallelSplitConvNode()