InfiniTensor/include/cuda/cuda_expand.h

#pragma once

#include "operators/unary.h"
#include "utils/small_array.h"
namespace infini {
void expandKernel(float *input, float *output, int nDims, int outputsize,
                  SmallArray inputShape, SmallArray outputShape);

}; // namespace infini