InfiniOps
Operator Library for Accelerators
Loading...
Searching...
No Matches
top_k_top_p_sampling_from_logits.h File Reference
#include <cassert>
#include <cstdint>
#include <optional>
#include <string>
#include "data_type.h"
#include "operator.h"
#include "tensor.h"

Go to the source code of this file.

Classes

class  infini::ops::TopKTopPSamplingFromLogits
 
struct  infini::ops::CacheKeyBuilder< TopKTopPSamplingFromLogits >
 

Namespaces

namespace  infini
 
namespace  infini::ops