MegEngine/imperative/python/src/tensor_utils.h

#pragma once

namespace mgb::imperative::python {

PyObject* dtype_promotion(PyObject* self, PyObject* const* args, size_t nargs);

PyObject* get_device(PyObject* self, PyObject* const* args, size_t nargs);

PyObject* make_shape_tuple(PyObject* self, PyObject* const* args, size_t nargs);

PyObject* getitem_cpp(PyObject* self, PyObject* const* args, size_t nargs);

PyObject* setitem_cpp(PyObject* self, PyObject* const* args, size_t nargs);

PyObject* split_cpp(PyObject* self, PyObject* const* args, size_t nargs);

PyObject* expand_dims_cpp(PyObject* self, PyObject* const* args, size_t nargs);

PyObject* squeeze_cpp(PyObject* self, PyObject* const* args, size_t nargs);

PyObject* transpose_cpp(PyObject* self, PyObject* const* args, size_t nargs);

PyObject* broadcast_cpp(PyObject* self, PyObject* const* args, size_t nargs);

PyObject* reshape_cpp(PyObject* self, PyObject* const* args, size_t nargs);

PyObject* adaptive_pool2d_cpp(PyObject* self, PyObject* const* args, size_t nargs);

PyObject* Const(PyObject* self, PyObject* const* args, size_t nargs);

PyObject* astype_cpp(PyObject* self, PyObject* const* args, size_t nargs);

PyObject* matmul_cpp(PyObject* self, PyObject* const* args, size_t nargs);

PyObject* batched_matmul_cpp(PyObject* self, PyObject* const* args, size_t nargs);

PyObject* convert_single_value_cpp(PyObject* self, PyObject* const* args, size_t nargs);

PyObject* convert_inputs_cpp(PyObject* self, PyObject* const* args, size_t nargs);

PyObject* astensor1d_cpp(PyObject* self, PyObject* const* args, size_t nargs);

PyObject* pixel_shuffle_cpp(PyObject* self, PyObject* const* args, size_t nargs);

}  // namespace mgb::imperative::python