PR Category

Custom Device

PR Types

Bug fixes

Description

• fix that TensorCopy changes the shape of dst Tensor in c_softmax_with_cross_entropy_op_xpu
• TensorCopy will resize the dst Tensor so we change it to memory::Copy.
• In the operator c_softmax_with_cross_entropy_op_xpu, the variablesoftmax_2d is flattened to 2d(e.g. [4096, 4000]), while the variable softmax may have higher dimensions (e.g. [1, 4096, 4000]). When using TensorCopy from softmax_2d to softmax, the dimensions of softmax will be changed.(e.g. from [1, 4096, 4000] to [4096, 4000]), leading to errors in the following operators or backward propogation.