Home
last modified time | relevance | path

Searched refs:py_outputs (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/framework/experimental/
Ddef_function.py54 py_outputs = nest.flatten(structured_outputs, expand_composites=True)
55 num_outputs = len(py_outputs)
57 finalized_f = ctx.Finalize(py_outputs)
/external/tensorflow/tensorflow/python/client/
Dtf_session_helper.cc354 std::vector<PyObject*>* py_outputs) { in TF_SessionRun_wrapper_helper() argument
356 DCHECK(py_outputs != nullptr); in TF_SessionRun_wrapper_helper()
357 DCHECK(py_outputs->empty()); in TF_SessionRun_wrapper_helper()
430 py_outputs->push_back(py_outputs_safe[i].release()); in TF_SessionRun_wrapper_helper()
440 std::vector<PyObject*>* py_outputs) { in TF_SessionRun_wrapper() argument
443 out_status, py_outputs); in TF_SessionRun_wrapper()
529 std::vector<PyObject*>* py_outputs) { in TF_SessionPRun_wrapper() argument
535 out_status, py_outputs); in TF_SessionPRun_wrapper()
Dtf_session_helper.h154 std::vector<PyObject*>* py_outputs);
186 std::vector<PyObject*>* py_outputs);
Dtf_session_wrapper.cc446 std::vector<PyObject*> py_outputs; in PYBIND11_MODULE() local
449 run_metadata, status.get(), &py_outputs); in PYBIND11_MODULE()
453 PyObject* result = PyList_New(py_outputs.size()); in PYBIND11_MODULE()
458 for (size_t i = 0; i < py_outputs.size(); ++i) { in PYBIND11_MODULE()
459 PyList_SET_ITEM(result, i, py_outputs.at(i)); in PYBIND11_MODULE()
498 std::vector<PyObject*> py_outputs; in PYBIND11_MODULE() local
500 outputs, status.get(), &py_outputs); in PYBIND11_MODULE()
503 PyObject* result = PyList_New(py_outputs.size()); in PYBIND11_MODULE()
508 for (size_t i = 0; i < py_outputs.size(); ++i) { in PYBIND11_MODULE()
509 PyList_SET_ITEM(result, i, py_outputs.at(i)); in PYBIND11_MODULE()