Searched refs:CalibrationWrapper (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/python/optimize/ |
| D | calibration_wrapper_pybind11.cc | 28 using tflite::calibration_wrapper::CalibrationWrapper;
38 py::class_<CalibrationWrapper>(m, "CalibrationWrapper") in PYBIND11_MODULE()
44 auto* wrapper = ::CalibrationWrapper::CreateWrapperCPPFromBuffer( in PYBIND11_MODULE()
52 [](CalibrationWrapper& self, py::handle& input_shapes, in PYBIND11_MODULE()
58 [](CalibrationWrapper& self, py::handle& input_shapes) { in PYBIND11_MODULE()
62 [](CalibrationWrapper& self, std::string signature_key) { in PYBIND11_MODULE()
66 [](CalibrationWrapper& self) { in PYBIND11_MODULE()
70 [](CalibrationWrapper& self, py::handle& input_value, in PYBIND11_MODULE()
76 [](CalibrationWrapper& self, py::handle& input_value) { in PYBIND11_MODULE()
80 [](CalibrationWrapper& self, int input_py_type, int output_py_type, in PYBIND11_MODULE()
[all …]
|
| D | calibration_wrapper.h | 56 class CalibrationWrapper {
59 static CalibrationWrapper* CreateWrapperCPPFromBuffer(
63 ~CalibrationWrapper();
111 CalibrationWrapper(
120 CalibrationWrapper(const CalibrationWrapper& rhs);
|
| D | calibration_wrapper.cc | 195 CalibrationWrapper::CalibrationWrapper( in CalibrationWrapper() function in tflite::calibration_wrapper::CalibrationWrapper
210 CalibrationWrapper::~CalibrationWrapper() {} in ~CalibrationWrapper()
212 PyObject* CalibrationWrapper::Prepare() { in Prepare()
219 PyObject* CalibrationWrapper::Prepare(std::string signature_key) { in Prepare()
233 PyObject* CalibrationWrapper::Prepare(PyObject* input_shapes, in Prepare()
274 PyObject* CalibrationWrapper::Prepare(PyObject* input_shapes) { in Prepare()
306 PyObject* CalibrationWrapper::FeedTensor(PyObject* input_value, in FeedTensor()
346 PyObject* CalibrationWrapper::FeedTensor(PyObject* input_value) { in FeedTensor()
378 PyObject* CalibrationWrapper::SetTensor(int index, PyObject* value, in SetTensor()
467 PyObject* CalibrationWrapper::SetTensor(int index, PyObject* value) { in SetTensor()
[all …]
|
| D | calibrator.py | 69 _calibration_wrapper.CalibrationWrapper(
|