|
|
@ -390,7 +390,7 @@ class DataNormKernel<platform::CPUDeviceContext, T>
|
|
|
|
}
|
|
|
|
}
|
|
|
|
default:
|
|
|
|
default:
|
|
|
|
PADDLE_THROW(platform::errors::InvalidArgument(
|
|
|
|
PADDLE_THROW(platform::errors::InvalidArgument(
|
|
|
|
"Unknown storage order: %d", data_layout));
|
|
|
|
"Unknown storage order: %d, please use NCHW or NHWC", data_layout));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
};
|
|
|
|
};
|
|
|
@ -701,7 +701,8 @@ class DataNormGradKernel<platform::CPUDeviceContext, T>
|
|
|
|
}
|
|
|
|
}
|
|
|
|
default:
|
|
|
|
default:
|
|
|
|
PADDLE_THROW(platform::errors::InvalidArgument(
|
|
|
|
PADDLE_THROW(platform::errors::InvalidArgument(
|
|
|
|
"Unknown storage order: %s", data_layout_str));
|
|
|
|
"Unknown storage order: %s, please use NCHW or NHWC",
|
|
|
|
|
|
|
|
data_layout_str));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
};
|
|
|
|
};
|
|
|
|