Commit ac199f2f authored by rocking's avatar rocking
Browse files

Refine declare type

parent 557d7aa3
...@@ -240,9 +240,9 @@ int main() ...@@ -240,9 +240,9 @@ int main()
const auto wei_element_op = WeiElementOp{}; const auto wei_element_op = WeiElementOp{};
const auto out_element_op = OutElementOp{0.5f, ActivationOp{}}; const auto out_element_op = OutElementOp{0.5f, ActivationOp{}};
using InLayout = decltype(ck::tensor_layout::convolution::GNHWC{}); using InLayout = ck::tensor_layout::convolution::GNHWC;
using WeiLayout = decltype(ck::tensor_layout::convolution::GKYXC{}); using WeiLayout = ck::tensor_layout::convolution::GKYXC;
using OutLayout = decltype(ck::tensor_layout::convolution::GNHWK{}); using OutLayout = ck::tensor_layout::convolution::GNHWK;
const auto in_g_n_c_wis_desc = const auto in_g_n_c_wis_desc =
ck::utils::conv::make_input_host_tensor_descriptor_g_n_c_wis_packed<InLayout>(conv_param); ck::utils::conv::make_input_host_tensor_descriptor_g_n_c_wis_packed<InLayout>(conv_param);
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment