singa-dev mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From GitBox <...@apache.org>
Subject [GitHub] [singa] chrishkchris edited a comment on issue #591: Dev branch cpu training problem (with conv and pool)
Date Tue, 18 Feb 2020 08:25:26 GMT
chrishkchris edited a comment on issue #591: Dev branch cpu training problem (with conv and
pool)
URL: https://github.com/apache/singa/issues/591#issuecomment-587300146
 
 
   @dcslin 
   
   I have tried to reorder the format before passing into conv, but seems I am not familiar
with the DNNL API and too difficult to debug
   See the reorder descripter I was trying and the new memory descriptor I added:
   https://github.com/chrishkchris/singa/blob/conv_reorder/src/model/operation/convolution.cc#L111
   
   From my understanding, the key concept and the step-by-step-prodcedure is:
   1. Create memory decriptor of tag::any format of conv2d for the input, weight, bias, output

   2. Create the conv primitive descriptor based on the memory decriptor created above
   3. Reorder the input and weight format to the tag::any format
   4. the reordered format input and weight can be passed into the conv for processing
   
   Could you help to make this work, thanks! 

----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
 
For queries about this service, please contact Infrastructure at:
users@infra.apache.org


With regards,
Apache Git Services

Mime
View raw message