forked from mindspore-Ecosystem/mindspore
!4115 runpackage sync C75B050 for mindspore r0.6
Merge pull request !4115 from HW_KK/r0.6
This commit is contained in:
commit
7d483cd09c
|
@ -1 +1 @@
|
|||
Subproject commit 103f2d1019dc50d781d7a964551d9f1f50b3b009
|
||||
Subproject commit 2cb83c8f4d8d3799081d016dd453a459e25c9879
|
|
@ -309,12 +309,7 @@ INPUT_MAP(SoftmaxCrossEntropyWithLogits) = {{1, INPUT_DESC(features)}, {2, INPUT
|
|||
ATTR_MAP(SoftmaxCrossEntropyWithLogits) = EMPTY_ATTR_MAP;
|
||||
OUTPUT_MAP(SoftmaxCrossEntropyWithLogits) = {{0, OUTPUT_DESC(loss)}, {1, OUTPUT_DESC(backprop)}};
|
||||
|
||||
// MeanGrad
|
||||
INPUT_MAP(MeanGrad) = {{1, INPUT_DESC(x)}};
|
||||
INPUT_ATTR_MAP(MeanGrad) = {{2, ATTR_DESC(mean_grad_output_shape_value, kOpFormat_NHWC,
|
||||
AnyTraits<std::vector<int64_t>>(), AnyTraits<int64_t>())}};
|
||||
ATTR_MAP(MeanGrad) = {{"mode", ATTR_DESC(mode, AnyTraits<int64_t>())}};
|
||||
|
||||
// SliceD
|
||||
INPUT_MAP(SliceD) = {{1, INPUT_DESC(x)}};
|
||||
INPUT_ATTR_MAP(SliceD) = {{2, ATTR_DESC(offsets, AnyTraits<int>(), AnyTraits<std::vector<int64_t>>())},
|
||||
{3, ATTR_DESC(size, AnyTraits<int>(), AnyTraits<std::vector<int64_t>>())}};
|
||||
|
@ -431,11 +426,6 @@ INPUT_MAP(TopK) = {{1, INPUT_DESC(x)}, {2, INPUT_DESC(k)}};
|
|||
ATTR_MAP(TopK) = {{"sorted", ATTR_DESC(sorted, AnyTraits<bool>())}};
|
||||
OUTPUT_MAP(TopK) = {{0, OUTPUT_DESC(values)}, {1, OUTPUT_DESC(indices)}};
|
||||
|
||||
// Multiply
|
||||
INPUT_MAP(Multiply) = {{1, INPUT_DESC(x)}, {2, INPUT_DESC(y)}};
|
||||
ATTR_MAP(Multiply) = EMPTY_ATTR_MAP;
|
||||
OUTPUT_MAP(Multiply) = {{0, OUTPUT_DESC(z)}};
|
||||
|
||||
// TileD
|
||||
INPUT_MAP(TileD) = {{1, INPUT_DESC(x)}};
|
||||
INPUT_ATTR_MAP(TileD) = {{2, ATTR_DESC(multiples, AnyTraits<int>(), AnyTraits<std::vector<int64_t>>())}};
|
||||
|
|
|
@ -70,8 +70,6 @@ DECLARE_OP_ADAPTER(AssignSub)
|
|||
DECLARE_OP_USE_OUTPUT(AssignSub)
|
||||
|
||||
DECLARE_OP_ADAPTER(ReduceMean)
|
||||
DECLARE_OP_ADAPTER(Multiply)
|
||||
DECLARE_OP_USE_OUTPUT(Multiply)
|
||||
|
||||
// ** Distributed Operations **
|
||||
DECLARE_OP_ADAPTER(HcomReduceScatter)
|
||||
|
@ -327,9 +325,6 @@ DECLARE_OP_USE_OUTPUT(MatMulV2)
|
|||
DECLARE_OP_ADAPTER(SoftmaxCrossEntropyWithLogits)
|
||||
DECLARE_OP_USE_OUTPUT(SoftmaxCrossEntropyWithLogits)
|
||||
|
||||
DECLARE_OP_ADAPTER(MeanGrad)
|
||||
DECLARE_OP_USE_INPUT_ATTR(MeanGrad)
|
||||
|
||||
DECLARE_OP_ADAPTER(Assign)
|
||||
DECLARE_OP_USE_OUTPUT(Assign)
|
||||
DECLARE_OP_ADAPTER(Constant)
|
||||
|
|
|
@ -293,12 +293,7 @@ INPUT_MAP(SoftmaxCrossEntropyWithLogits) = {{1, INPUT_DESC(features)}, {2, INPUT
|
|||
ATTR_MAP(SoftmaxCrossEntropyWithLogits) = EMPTY_ATTR_MAP;
|
||||
OUTPUT_MAP(SoftmaxCrossEntropyWithLogits) = {{0, OUTPUT_DESC(loss)}, {1, OUTPUT_DESC(backprop)}};
|
||||
|
||||
// MeanGrad
|
||||
INPUT_MAP(MeanGrad) = {{1, INPUT_DESC(x)}};
|
||||
INPUT_ATTR_MAP(MeanGrad) = {{2, ATTR_DESC(mean_grad_output_shape_value, kOpFormat_NHWC,
|
||||
AnyTraits<std::vector<int64_t>>(), AnyTraits<int64_t>())}};
|
||||
ATTR_MAP(MeanGrad) = {{"mode", ATTR_DESC(mode, AnyTraits<int64_t>())}};
|
||||
|
||||
// SliceD
|
||||
INPUT_MAP(SliceD) = {{1, INPUT_DESC(x)}};
|
||||
INPUT_ATTR_MAP(SliceD) = {{2, ATTR_DESC(offsets, AnyTraits<int>(), AnyTraits<std::vector<int64_t>>())},
|
||||
{3, ATTR_DESC(size, AnyTraits<int>(), AnyTraits<std::vector<int64_t>>())}};
|
||||
|
@ -415,11 +410,6 @@ INPUT_MAP(TopK) = {{1, INPUT_DESC(x)}, {2, INPUT_DESC(k)}};
|
|||
ATTR_MAP(TopK) = {{"sorted", ATTR_DESC(sorted, AnyTraits<bool>())}};
|
||||
OUTPUT_MAP(TopK) = {{0, OUTPUT_DESC(values)}, {1, OUTPUT_DESC(indices)}};
|
||||
|
||||
// Multiply
|
||||
INPUT_MAP(Multiply) = {{1, INPUT_DESC(x)}, {2, INPUT_DESC(y)}};
|
||||
ATTR_MAP(Multiply) = EMPTY_ATTR_MAP;
|
||||
OUTPUT_MAP(Multiply) = {{0, OUTPUT_DESC(z)}};
|
||||
|
||||
// TileD
|
||||
INPUT_MAP(TileD) = {{1, INPUT_DESC(x)}};
|
||||
INPUT_ATTR_MAP(TileD) = {{2, ATTR_DESC(multiples, AnyTraits<int>(), AnyTraits<std::vector<int64_t>>())}};
|
||||
|
|
Loading…
Reference in New Issue