Home
last modified time | relevance | path

Searched refs:PyoOrThrow (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/lite/python/interpreter_wrapper/
Dinterpreter_wrapper_pybind11.cc86 return tensorflow::PyoOrThrow(self.AllocateTensors()); in PYBIND11_MODULE()
90 return tensorflow::PyoOrThrow(self.Invoke()); in PYBIND11_MODULE()
94 return tensorflow::PyoOrThrow(self.InputIndices()); in PYBIND11_MODULE()
98 return tensorflow::PyoOrThrow(self.OutputIndices()); in PYBIND11_MODULE()
102 return tensorflow::PyoOrThrow( in PYBIND11_MODULE()
109 return tensorflow::PyoOrThrow(self.TensorType(i)); in PYBIND11_MODULE()
113 return tensorflow::PyoOrThrow(self.TensorSize(i)); in PYBIND11_MODULE()
117 return tensorflow::PyoOrThrow(self.TensorSizeSignature(i)); in PYBIND11_MODULE()
121 return tensorflow::PyoOrThrow(self.TensorSparsityParameters(i)); in PYBIND11_MODULE()
126 return tensorflow::PyoOrThrow(self.TensorQuantization(i)); in PYBIND11_MODULE()
[all …]
/external/tensorflow/tensorflow/lite/python/optimize/
Dcalibration_wrapper_pybind11.cc30 return tensorflow::PyoOrThrow(AddIntermediateTensors(data.ptr())); in PYBIND11_MODULE()
38 return tensorflow::PyoOrThrow(self.Prepare(input_shapes.ptr())); in PYBIND11_MODULE()
42 return tensorflow::PyoOrThrow(self.Prepare()); in PYBIND11_MODULE()
46 return tensorflow::PyoOrThrow(self.FeedTensor(input_value.ptr())); in PYBIND11_MODULE()
52 return tensorflow::PyoOrThrow( in PYBIND11_MODULE()
59 return tensorflow::PyoOrThrow( in PYBIND11_MODULE()
66 return tensorflow::PyoOrThrow( in PYBIND11_MODULE()
71 return tensorflow::PyoOrThrow(self.Calibrate()); in PYBIND11_MODULE()
/external/tensorflow/tensorflow/python/lite/
Dtoco_python_api_wrapper.cc29 return tensorflow::PyoOrThrow(toco::TocoConvert( in PYBIND11_MODULE()
52 return tensorflow::PyoOrThrow(toco::TocoGetPotentiallySupportedOps()); in PYBIND11_MODULE()
61 return tensorflow::PyoOrThrow(toco::MlirQuantizeModel( in PYBIND11_MODULE()
74 return tensorflow::PyoOrThrow( in PYBIND11_MODULE()
84 return tensorflow::PyoOrThrow( in PYBIND11_MODULE()
/external/tensorflow/tensorflow/python/
Dtfe_wrapper.cc240 return tensorflow::PyoOrThrow(packed_tensor); in InputTFE_OutputTensorHandles()
267 return tensorflow::PyoOrThrow(output_list); in InputTFE_OutputTensorHandles()
282 return tensorflow::PyoOrThrow(result); in InputTFE_OutputTensorHandles()
520 return tensorflow::PyoOrThrow(TFE_Py_RegisterExceptionClass(e.ptr())); in PYBIND11_MODULE()
523 return tensorflow::PyoOrThrow( in PYBIND11_MODULE()
625 return tensorflow::PyoOrThrow(tensorflow::OutputTFE_Context(context)); in PYBIND11_MODULE()
829 return tensorflow::PyoOrThrow(list); in PYBIND11_MODULE()
831 return tensorflow::PyoOrThrow(output_pyo); in PYBIND11_MODULE()
834 return tensorflow::PyoOrThrow(TFE_Py_InitEagerTensor(o.ptr())); in PYBIND11_MODULE()
842 return tensorflow::PyoOrThrow(TFE_Py_RegisterJVPFunction(o.ptr())); in PYBIND11_MODULE()
[all …]
/external/tensorflow/tensorflow/python/util/
Dutil_wrapper.cc31 return tensorflow::PyoOrThrow( in PYBIND11_MODULE()
35 return tensorflow::PyoOrThrow( in PYBIND11_MODULE()
124 return tensorflow::PyoOrThrow( in PYBIND11_MODULE()
205 return tensorflow::PyoOrThrow( in PYBIND11_MODULE()
228 return tensorflow::PyoOrThrow( in PYBIND11_MODULE()
288 return tensorflow::PyoOrThrow( in PYBIND11_MODULE()
Dnest_wrapper.cc30 return tensorflow::PyoOrThrow(tensorflow::FlattenDictItems(dict.ptr())); in PYBIND11_MODULE()
Dpy_checkpoint_reader_wrapper.cc115 return tensorflow::PyoOrThrow( in CheckpointReader_GetTensor()
/external/tensorflow/tensorflow/lite/testing/
Dstring_util_wrapper.cc31 return tensorflow::PyoOrThrow( in PYBIND11_MODULE()
/external/tensorflow/tensorflow/python/training/
Dquantize_training_wrapper.cc44 return tensorflow::PyoOrThrow( in PYBIND11_MODULE()
/external/tensorflow/tensorflow/python/lib/core/
Dpybind11_lib.h51 inline py::object PyoOrThrow(PyObject* ptr) { in PyoOrThrow() function
/external/tensorflow/tensorflow/python/eager/
Dcustom_device_testutil.cc67 return tensorflow::PyoOrThrow( in PYBIND11_MODULE()
/external/tensorflow/tensorflow/python/distribute/parallel_device/
Dpywrap_parallel_device.cc67 return tensorflow::PyoOrThrow( in PYBIND11_MODULE()
/external/tensorflow/tensorflow/python/client/
Dtf_session_wrapper.cc380 return tensorflow::PyoOrThrow(result); in PYBIND11_MODULE()
462 return tensorflow::PyoOrThrow(result); in PYBIND11_MODULE()
512 return tensorflow::PyoOrThrow(result); in PYBIND11_MODULE()
670 return tensorflow::PyoOrThrow(PyBytes_FromStringAndSize( in PYBIND11_MODULE()
728 return tensorflow::PyoOrThrow(PyBytes_FromStringAndSize( in PYBIND11_MODULE()