blob: d1c03a9a838534b1ca2945c8b2488619a2fc655d [file] [log] [blame]
#pragma once
#ifdef BUILD_NAMEDTENSOR
#include <torch/csrc/python_headers.h>
#include <ATen/Dimname.h>
at::Dimname THPDimname_parse(PyObject* obj);
bool THPUtils_checkDimname(PyObject* obj);
bool THPUtils_checkDimnameList(PyObject* obj);
#endif