Packages

extorch

0.4.0
0.4.0 0.3.0 0.2.0 0.1.0-pre0

Production ML model serving on the BEAM. Serve PyTorch models faster than Python with pre-compiled graph execution, AOTI compiled inference, and OTP fault tolerance.

HexDocs

Current section

Files

Jump to
Readme
4 Versions
3 Dependencies
Dependants
Files
Activity
extorch native extorch include comparison.h
Raw

native/extorch/include/comparison.h 



#include "common.h"

#include "utils.h"



bool allclose(

        const std::shared_ptr<CrossTensor> &input,

        const std::shared_ptr<CrossTensor> &other,

        double rtol, double atol, bool equal_nan);



std::shared_ptr<CrossTensor> argsort(

        const std::shared_ptr<CrossTensor> &input,

        int64_t dim,

        bool descending,

        bool stable);



SortResult sort(

        const std::shared_ptr<CrossTensor> &input,

        int64_t dim,

        bool descending,

        bool stable,

        SortResult out_r);



std::shared_ptr<CrossTensor> eq(

        const std::shared_ptr<CrossTensor> &input,

        const std::shared_ptr<CrossTensor> &other,

        TensorOut out);



bool equal(const std::shared_ptr<CrossTensor> &input,

           const std::shared_ptr<CrossTensor> &other);



std::shared_ptr<CrossTensor> ge(

        const std::shared_ptr<CrossTensor> &input,

        const std::shared_ptr<CrossTensor> &other,

        TensorOut out);



std::shared_ptr<CrossTensor> gt(

        const std::shared_ptr<CrossTensor> &input,

        const std::shared_ptr<CrossTensor> &other,

        TensorOut out);



std::shared_ptr<CrossTensor> le(

        const std::shared_ptr<CrossTensor> &input,

        const std::shared_ptr<CrossTensor> &other,

        TensorOut out);



std::shared_ptr<CrossTensor> lt(

        const std::shared_ptr<CrossTensor> &input,

        const std::shared_ptr<CrossTensor> &other,

        TensorOut out);



std::shared_ptr<CrossTensor> ne(

        const std::shared_ptr<CrossTensor> &input,

        const std::shared_ptr<CrossTensor> &other,

        TensorOut out);



std::shared_ptr<CrossTensor> isclose(

        const std::shared_ptr<CrossTensor> &input,

        const std::shared_ptr<CrossTensor> &other,

        double rtol, double atol, bool equal_nan);



std::shared_ptr<CrossTensor> isfinite(const std::shared_ptr<CrossTensor> &input);

std::shared_ptr<CrossTensor> isinf(const std::shared_ptr<CrossTensor> &input);

std::shared_ptr<CrossTensor> isposinf(const std::shared_ptr<CrossTensor> &input);

std::shared_ptr<CrossTensor> isneginf(const std::shared_ptr<CrossTensor> &input);

std::shared_ptr<CrossTensor> isnan(const std::shared_ptr<CrossTensor> &input);

std::shared_ptr<CrossTensor> isreal(const std::shared_ptr<CrossTensor> &input);



std::shared_ptr<CrossTensor> isin(

        const std::shared_ptr<CrossTensor> &elements,

        const std::shared_ptr<CrossTensor> &test_elements,

        bool assume_unique, bool invert);



SortResult kthvalue(

        const std::shared_ptr<CrossTensor> &input,

        int64_t k,

        int64_t dim,

        bool keepdim,

        SortResult out_r);



std::shared_ptr<CrossTensor> maximum(

        const std::shared_ptr<CrossTensor> &input,

        const std::shared_ptr<CrossTensor> &other,

        TensorOut out);



std::shared_ptr<CrossTensor> minimum(

        const std::shared_ptr<CrossTensor> &input,

        const std::shared_ptr<CrossTensor> &other,

        TensorOut out);



std::shared_ptr<CrossTensor> fmax(

        const std::shared_ptr<CrossTensor> &input,

        const std::shared_ptr<CrossTensor> &other,

        TensorOut out);



std::shared_ptr<CrossTensor> fmin(

        const std::shared_ptr<CrossTensor> &input,

        const std::shared_ptr<CrossTensor> &other,

        TensorOut out);





SortResult topk(

        const std::shared_ptr<CrossTensor> &input,

        int64_t k,

        int64_t dim,

        bool largest,

        bool sorted,

        SortResult out_r

);



std::shared_ptr<CrossTensor> msort(

        const std::shared_ptr<CrossTensor> &input,

        TensorOut out);