Template Function marian::gpu::gCrossEntropyPick

• Defined in File tensor_operators.cu

Function Documentation

template<typename T, typename AccType = float>__global__ void marian::gpu::gCrossEntropyPick(AccType * out, const functional::Shape outShape, const T * in, const functional::Shape inShape, const IndexType * pick, AccType labelSmoothingAlpha = AccType(0.f))