InfiniTensor/include/cuda/cuda_clip.h

10 lines
192 B
C++

#pragma once
#include "operators/unary.h"
namespace infini {
void clip_kernel(float *input, float *output, int num, float minValue,
float maxValue);
}; // namespace infini