|
|
|
@ -808,11 +808,11 @@ ATTR_MAP(DepthwiseConv2DBackpropFilterD) = {
|
|
|
|
|
};
|
|
|
|
|
OUTPUT_MAP(DepthwiseConv2DBackpropFilterD) = {{0, OUTPUT_DESC(filter_grad)}};
|
|
|
|
|
|
|
|
|
|
// MatMul
|
|
|
|
|
INPUT_MAP(MatMul) = {{1, INPUT_DESC(x1)}, {2, INPUT_DESC(x2)}};
|
|
|
|
|
ATTR_MAP(MatMul) = {{"transpose_a", ATTR_DESC(transpose_x1, AnyTraits<bool>())},
|
|
|
|
|
{"transpose_b", ATTR_DESC(transpose_x2, AnyTraits<bool>())}};
|
|
|
|
|
OUTPUT_MAP(MatMul) = {{0, OUTPUT_DESC(y)}};
|
|
|
|
|
// MatMulV2
|
|
|
|
|
INPUT_MAP(MatMulV2) = {{1, INPUT_DESC(x1)}, {2, INPUT_DESC(x2)}};
|
|
|
|
|
ATTR_MAP(MatMulV2) = {{"transpose_a", ATTR_DESC(transpose_x1, AnyTraits<bool>())},
|
|
|
|
|
{"transpose_b", ATTR_DESC(transpose_x2, AnyTraits<bool>())}};
|
|
|
|
|
OUTPUT_MAP(MatMulV2) = {{0, OUTPUT_DESC(y)}};
|
|
|
|
|
|
|
|
|
|
// Merge
|
|
|
|
|
INPUT_MAP(Merge) = EMPTY_INPUT_MAP;
|
|
|
|
|