xla

72bfe5d2 - Fix depthwise conv (#973)

Fix depthwise conv (#973) * remove 2d in function name since it's general now * use XLA logic of depthwise in forward * fix depthwise conv * add note about lowering convolution * address comments