forked from mindspore-Ecosystem/mindspore
!4238 update run to c75b100
Merge pull request !4238 from guozhijian/update_run_c75b100
This commit is contained in:
commit
3ea803aa7d
|
@ -1 +1 @@
|
|||
Subproject commit 6d12411003164d88eaed62e1ead33761cbfa15ef
|
||||
Subproject commit 622af6c1c50034bea5a08bd409c5a410782bfe53
|
|
@ -309,12 +309,6 @@ 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>())}};
|
||||
|
||||
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 +425,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,6 @@ 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>())}};
|
||||
|
||||
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 +409,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