Searched refs:SetTensor (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | copy_to_device_node.h | 65 return dst_->SetTensor(std::move(tensor), dstd_); in Run()
|
| D | tensor_handle_data.h | 58 Status SetTensor(tensorflow::Tensor&& t);
|
| D | tensor_handle_data.cc | 84 Status LocalTensorHandleData::SetTensor(tensorflow::Tensor&& t) { in SetTensor() function in tensorflow::LocalTensorHandleData
|
| D | tensor_handle.h | 218 Status SetTensor(tensorflow::Tensor&& tensor, const Device* d);
|
| D | tensor_handle.cc | 927 Status TensorHandle::SetTensor(tensorflow::Tensor&& t, const Device* d) { in SetTensor() function in tensorflow::TensorHandle
938 return data.SetTensor(std::move(t)); in SetTensor()
948 return mirror.SetTensor(std::move(t)); in SetTensor()
|
| D | execute.cc | 1028 TF_RETURN_IF_ERROR(retvals[i]->SetTensor( in GetKernelOutputs()
|
| /external/tensorflow/tensorflow/lite/python/optimize/ |
| D | calibration_wrapper.h | 94 PyObject* SetTensor(int index, PyObject* value);
|
| D | calibration_wrapper.cc | 227 if (!SetTensor(input_tensor_idx, input)) { in FeedTensor()
236 PyObject* CalibrationWrapper::SetTensor(int index, PyObject* value) { in SetTensor() function in tflite::calibration_wrapper::CalibrationWrapper
|
| /external/tensorflow/tensorflow/lite/python/interpreter_wrapper/ |
| D | interpreter_wrapper.h | 88 PyObject* SetTensor(int i, PyObject* value);
|
| D | interpreter_wrapper_pybind11.cc | 138 return tensorflow::PyoOrThrow(self.SetTensor(i, value.ptr())); in PYBIND11_MODULE()
|
| D | interpreter_wrapper.cc | 417 PyObject* InterpreterWrapper::SetTensor(int i, PyObject* value) { in SetTensor() function in tflite::interpreter_wrapper::InterpreterWrapper
607 return SetTensor(input->second, value); in SetInputTensorFromSignatureDefName()
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | interpreter.py | 499 self._interpreter.SetTensor(tensor_index, value)
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_copy_node.cc | 302 captured_state_->dst()->SetTensor(std::move(outputs[0]), recv_device); in StartRecv()
|