operator_call_instantiations.h File Reference

#include <cstdint>
#include <optional>
#include <vector>
#include "config.h"
#include "handle.h"
#include "operator.h"
#include "base/causal_softmax_infinilm.h"
#include "base/rms_norm.h"
#include "base/cast.h"
#include "base/linear.h"
#include "base/swiglu.h"
#include "base/silu.h"
#include "base/top_k_top_p_sampler.h"
#include "base/silu_and_mul.h"
#include "base/relu.h"
#include "base/add.h"
#include "base/causal_softmax.h"
#include "base/mul.h"
#include "base/top_k_top_p_sample_infinilm.h"
#include "base/cat.h"
#include "base/top_k_top_p_sampling_from_logits.h"
#include "base/gemm.h"

Go to the source code of this file.

Namespaces
namespace	infini
namespace	infini::ops

Functions
template void	infini::ops::Tensor (const Tensor &input, const Tensor &weight, const std::optional< Tensor > &bias, const Tensor &out) const
template void	infini::ops::Tensor (const Tensor &logits, const std::optional< Tensor > &k, const std::optional< Tensor > &p, const Tensor &out) const
template void	infini::ops::Tensor (const Tensor &logits, const std::optional< Tensor > &k, const std::optional< Tensor > &p, const uint64_t &seed, const uint64_t &offset, const Tensor &out) const
template void	infini::ops::Tensor (const Tensor &logits, const Tensor &top_k, const Tensor &top_p, const std::optional< Tensor > &indices, const std::string &filter_apply_order, const bool &deterministic, const bool &check_nan, const std::optional< int64_t > &seed, const std::optional< int64_t > &offset, const Tensor &out) const
template void	infini::ops::Tensor (const Tensor &a, const Tensor &b, const std::optional< Tensor > &c, const std::optional< float > &alpha, const std::optional< float > &beta, const std::optional< int > &trans_a, const std::optional< int > &trans_b, const Tensor &y) const

Variables
template void	infini::ops::uint64_t
template void	infini::ops::bool