Closed weiyanlin117 closed 1 year ago
test passed only support weights gradient and output gradient, on bias gradient yet, Since I notice pytorch doesn't implement bias gradient, I treat bias gradient as optional? I can spend time if it is needed.
Have you tried NHWC layout (with weights potentially in NCHW layout)?
NCHW works as same as previous
Probably need a regen.
normal test passed. commented out dbias test passed.
test passed only support weights gradient and output gradient, on bias gradient yet, Since I notice pytorch doesn't implement bias gradient, I treat bias gradient as optional? I can spend time if it is needed.