forked from jiuyuan/InfiniTensor
13 lines
390 B
C++
13 lines
390 B
C++
#pragma once
|
|
|
|
#include "operators/unary.h"
|
|
#include "utils/small_array.h"
|
|
namespace infini {
|
|
void expandKernel(int dType, void *input, void *output, int nDims,
|
|
int outputsize, SmallArray inputShape,
|
|
SmallArray outputShape);
|
|
|
|
void expandRowKernel(int dType, void *input, void *output, int n_rows,
|
|
int row_len);
|
|
}; // namespace infini
|