提交 109e2e9b 编写于 作者: L limingqi107

modify the format info of tensorAdd

上级 1b6f628b
......@@ -56,9 +56,9 @@ class BroadcastOpGpuKernel : public GpuKernel {
}
bool Init(const CNodePtr &kernel_node) override {
GetOpType(kernel_node);
auto shape1 = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 0);
auto shape2 = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 1);
auto shape3 = AnfAlgo::GetOutputInferShape(kernel_node, 0);
auto shape1 = AnfAlgo::GetInputDeviceShape(kernel_node, 0);
auto shape2 = AnfAlgo::GetInputDeviceShape(kernel_node, 1);
auto shape3 = AnfAlgo::GetOutputDeviceShape(kernel_node, 0);
need_broadcast_ = IsBroadcast(shape1, shape2);
if (need_broadcast_ && shape1.size() > 7) {
MS_LOG(EXCEPTION) << "Broadcast operation not support dim greater than 7";
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册