fix complite bug in clang

pull/230/head
chenzomi 5 years ago
parent aaa8d9ed71
commit e09f220f17

@ -174,14 +174,14 @@ OPERATOR_ONNX_CONVERT_DEFINE(Sigmoid, Sigmoid, OpNameInfo())
OPERATOR_ONNX_CONVERT_DEFINE(Flatten, Flatten, OpNameInfo())
OPERATOR_ONNX_CONVERT_DEFINE(Squeeze, Squeeze,
OpNameInfo().Attr("axis", "axes", onnx::AttributeProto_AttributeType_INTS,
SetAttrTupleValueToProto))
SetAttrTupleValueToProto<0>))
OPERATOR_ONNX_CONVERT_DEFINE(
Conv2D, Conv,
OpNameInfo()
.Attr("dilation", "dilations", onnx::AttributeProto_AttributeType_INTS, SetAttrValueToProto<Int32Imm, 2>)
.Attr("group", "group", onnx::AttributeProto_AttributeType_INT, SetAttrValueToProto<Int32Imm>)
.Attr("kernel_size", "kernel_shape", onnx::AttributeProto_AttributeType_INTS, SetAttrTupleValueToProto)
.Attr("kernel_size", "kernel_shape", onnx::AttributeProto_AttributeType_INTS, SetAttrTupleValueToProto<0>)
.Attr("pad_mode", "auto_pad", onnx::AttributeProto_AttributeType_STRING,
[](ValuePtr value, onnx::AttributeProto_AttributeType, onnx::AttributeProto* const attr_proto,
const PrimitivePtr& prim) {

@ -15,7 +15,7 @@
*/
#ifndef MINDSPORE_CCSRC_OPTIMIZER_IRPASS_INCORPORATE_GETITEM_H_
#define MINDSPORE_CCSRC_OPTIMIZER_IRPASS_INCORPORATE_GETITEM_H__
#define MINDSPORE_CCSRC_OPTIMIZER_IRPASS_INCORPORATE_GETITEM_H_
#include <vector>
#include <algorithm>

@ -236,7 +236,7 @@ OperatorVector CreateMirrorOps(const std::string& group_name, size_t dev_num) {
OperatorName operator_name = MIRROR_OPERATOR;
ValuePtr attr0_value = MakeValue(group_name);
ValuePtr attr1_value = MakeValue(dev_num);
ValuePtr attr1_value = MakeValue(SizeToInt(dev_num));
ValuePtr attr2_value = MakeValue(mean_flag);
Attr attr0 = std::make_pair(GROUP, attr0_value);

@ -22,6 +22,7 @@
#include <string>
#include <sstream>
#include <memory>
#include "./overload.h"
#include "./securec.h"
#ifdef USE_GLOG
#include "glog/logging.h"

Loading…
Cancel
Save