Home
last modified time | relevance | path

Searched refs:SetTensor (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcopy_to_device_node.h46 return dst_->SetTensor(std::move(tensor)); in Run()
Dtensor_handle.h161 Status SetTensor(tensorflow::Tensor&& tensor);
Dtensor_handle.cc561 Status TensorHandle::SetTensor(tensorflow::Tensor&& tensor) { in SetTensor() function in tensorflow::TensorHandle
Dexecute.cc1001 TF_RETURN_IF_ERROR(retvals[i]->SetTensor(std::move(outputs[i]))); in EagerKernelExecute()
/external/tensorflow/tensorflow/lite/python/optimize/
Dcalibration_wrapper.h85 PyObject* SetTensor(int index, PyObject* value);
Dcalibration_wrapper.cc142 if (!SetTensor(input_tensor_idx, input)) { in FeedTensor()
151 PyObject* CalibrationWrapper::SetTensor(int index, PyObject* value) { in SetTensor() function in tflite::calibration_wrapper::CalibrationWrapper
/external/tensorflow/tensorflow/lite/python/interpreter_wrapper/
Dinterpreter_wrapper.h76 PyObject* SetTensor(int i, PyObject* value);
Dinterpreter_wrapper.cc362 PyObject* InterpreterWrapper::SetTensor(int i, PyObject* value) { in SetTensor() function in tflite::interpreter_wrapper::InterpreterWrapper
/external/tensorflow/tensorflow/lite/python/
Dinterpreter.py401 self._interpreter.SetTensor(tensor_index, value)
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dremote_copy_node.cc278 status = captured_state_->dst()->SetTensor(std::move(outputs[0])); in StartRecv()