Searched refs:indicesH (Results 1 – 2 of 2) sorted by relevance
4605 int64_t idx = indicesH.raw(posIn); in fwdSparseToDenseMaskInst()4620 assert(posIn == indicesH.dims()[0] && in fwdSparseToDenseMaskInst()5284 auto indicesH = getTensor(indices)->getHandle<T>(); in fwdNonMaxSuppressionInstImpl() local5359 indicesH.at({outPutBoxIndex}) = priorBox.second; in fwdNonMaxSuppressionInstImpl()5362 indicesH.at({outPutBoxIndex, 0}) = batchIndex; in fwdNonMaxSuppressionInstImpl()5363 indicesH.at({outPutBoxIndex, 1}) = classIndex; in fwdNonMaxSuppressionInstImpl()5381 minBox.boxIndex = indicesH.at({outPutBoxIndex - 1}); in fwdNonMaxSuppressionInstImpl()5390 indicesH.at({outPutBoxIndex}) = minBox.boxIndex; in fwdNonMaxSuppressionInstImpl()5392 indicesH.at({outPutBoxIndex, 0}) = minBox.batchIndex; in fwdNonMaxSuppressionInstImpl()5393 indicesH.at({outPutBoxIndex, 1}) = minBox.classIndex; in fwdNonMaxSuppressionInstImpl()[all …]
1772 auto indicesH = ctx_.allocate(indices)->getHandle<int32_t>();1773 indicesH = inputIds;1788 indicesH = inputIds;