Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/signal/
H A Dtest_util.py68 def evaluate_tflite_model(tflite_model, input_ndarrays): argument
88 if len(input_details) != len(input_ndarrays):
91 input_ndarrays, input_details, output_details))
92 for input_tensor, data in zip(input_details, input_ndarrays):
/aosp_15_r20/external/tensorflow/tensorflow/python/client/
H A Dtf_session_helper.cc350 const std::vector<PyObject*>& input_ndarrays, in TF_SessionRun_wrapper_helper() argument
356 DCHECK_EQ(inputs.size(), input_ndarrays.size()); in TF_SessionRun_wrapper_helper()
379 for (PyObject* ndarray : input_ndarrays) { in TF_SessionRun_wrapper_helper()
437 const std::vector<PyObject*>& input_ndarrays, in TF_SessionRun_wrapper() argument
443 input_ndarrays, outputs, targets, run_metadata, in TF_SessionRun_wrapper()
527 const std::vector<PyObject*>& input_ndarrays, in TF_SessionPRun_wrapper() argument
534 inputs, input_ndarrays, outputs, targets, in TF_SessionPRun_wrapper()
H A Dtf_session_helper.h150 const std::vector<PyObject*>& input_ndarrays,
183 const std::vector<PyObject*>& input_ndarrays,
H A Dtf_session_wrapper.cc426 std::vector<PyObject*> input_ndarrays; in PYBIND11_MODULE() local
448 input_ndarrays.push_back(value); in PYBIND11_MODULE()
455 input_ndarrays, outputs, targets, in PYBIND11_MODULE()
478 std::vector<PyObject*> input_ndarrays; in PYBIND11_MODULE() local
500 input_ndarrays.push_back(value); in PYBIND11_MODULE()
506 tensorflow::TF_SessionPRun_wrapper(session, handle, inputs, input_ndarrays, in PYBIND11_MODULE()