Template Function marian::gpu::gSelect

• Defined in File tensor_operators.cu

Function Documentation

template<typename T>__global__ void marian::gpu::gSelect(T * out, functional::Shape outShape, const T * in, const functional::Shape inShape, int axis, const IndexType * d_indices, const functional::Shape idxShape)