// Copyright (c) 2021 Graphcore Ltd. All rights reserved.
#ifndef POPART_PYTHON_POPART__INTERNAL_IR_BINDINGS_OP_PRINTTENSOR_HPP_
#define POPART_PYTHON_POPART__INTERNAL_IR_BINDINGS_OP_PRINTTENSOR_HPP_

#include <pybind11/pybind11.h>

namespace py = pybind11;

namespace popart {
namespace _internal {
namespace ir {
namespace op {

/**
 * Add bindings for the PrintTensor op.
 **/
void bindPrintTensor(py::module &m);
} // namespace op

/**
 * Add bindings for `poplar::PrintTensorFmt` class to pybind module.
 **/
void bindPrintTensorFmt(py::module &m);

/**
 * Add bindings for `poplar::bindFloatFormat` class to pybind module.
 **/
void bindFloatFormat(py::module &m);

} // namespace ir
} // namespace _internal
} // namespace popart

#endif // POPART_PYTHON_POPART__INTERNAL_IR_BINDINGS_OP_PRINTTENSOR_HPP_