forked from mindspore-Ecosystem/mindspore
!230 fix clang complier bug
Merge pull request !230 from SanjayChan/master
This commit is contained in:
commit
5ec3c8a215
|
@ -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…
Reference in New Issue