• F
    support channel last in BatchNorm*d · 2e845182
    Feiyu Chan 提交于
    1. support channel last in BatchNorm*d (#27875)
    2. fix a bug in batch_norm_op cuda kernel by extracting ResizeToChannelFist(Last), TransToChannelFirst(Last) to operators/layer_utils.h
    2e845182
test_batch_norm_op_v2.py 9.3 KB