| /aosp_15_r20/external/tensorflow/tensorflow/lite/python/optimize/ |
| H A D | calibration_wrapper.h | 122 PyObject* SetTensor(int index, PyObject* value);
123 PyObject* SetTensor(int index, PyObject* value, std::string signature_key);
|
| H A D | calibration_wrapper.cc | 337 if (!SetTensor(input_tensor_idx, input, signature_key)) { in FeedTensor()
369 if (!SetTensor(input_tensor_idx, input)) { in FeedTensor()
378 PyObject* CalibrationWrapper::SetTensor(int index, PyObject* value, in SetTensor() function in tflite::calibration_wrapper::CalibrationWrapper
467 PyObject* CalibrationWrapper::SetTensor(int index, PyObject* value) { in SetTensor() function in tflite::calibration_wrapper::CalibrationWrapper
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/python/ |
| H A D | interpreter.py | 254 self._interpreter_wrapper.SetTensor(self._inputs[input_name], value,
697 self._interpreter.SetTensor(tensor_index, value)
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/ |
| H A D | copy_to_device_node.h | 66 return dst_->SetTensor(std::move(tensor), dstd_); in Run()
|
| H A D | tensor_handle_data.h | 58 Status SetTensor(tensorflow::Tensor&& t);
|
| H A D | tensor_handle_data.cc | 84 Status LocalTensorHandleData::SetTensor(tensorflow::Tensor&& t) { in SetTensor() function in tensorflow::LocalTensorHandleData
|
| H A 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()
|
| H A D | tensor_handle.h | 224 Status SetTensor(tensorflow::Tensor&& tensor, const Device* d);
|
| H A D | execute.cc | 1739 TF_RETURN_IF_ERROR(retvals[i]->SetTensor( in GetKernelOutputs()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/python/interpreter_wrapper/ |
| H A D | interpreter_wrapper.h | 87 PyObject* SetTensor(int i, PyObject* value, int subgraph_index);
|
| H A D | interpreter_wrapper_pybind11.cc | 162 self.SetTensor(i, value.ptr(), subgraph_index)); in PYBIND11_MODULE()
|
| H A D | interpreter_wrapper.cc | 488 PyObject* InterpreterWrapper::SetTensor(int i, PyObject* value, in SetTensor() function in tflite::interpreter_wrapper::InterpreterWrapper
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| H A D | inference_context.h | 128 absl::Status SetTensor(const ValueId& tensor_id,
|
| H A D | inference_context.cc | 771 absl::Status InferenceContext::SetTensor(const ValueId& tensor_id, in SetTensor() function in tflite::gpu::metal::InferenceContext
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/core/ |
| H A D | tensor.h | 88 void SetTensor(at::Tensor tensor);
|
| H A D | tensor.cpp | 271 void LazyTensor::SetTensor(at::Tensor tensor) { in SetTensor() function in torch::lazy::LazyTensor
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| H A D | inference_context.h | 111 absl::Status SetTensor(const ValueId& tensor_id, Tensor* tensor_ptr);
|
| H A D | inference_context.cc | 699 absl::Status InferenceContext::SetTensor(const ValueId& tensor_id, in SetTensor() function in tflite::gpu::cl::InferenceContext
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/ |
| H A D | metal_delegate.mm | 217 RETURN_IF_ERROR(inference_context_.SetTensor(input.id, in_out_tensors_[input.id].get()));
228 inference_context_.SetTensor(output.id, in_out_tensors_[output.id].get()));
466 inference_context_.SetTensor(external_tensor.first, external_tensor.second.get()));
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| H A D | remote_copy_node.cc | 311 captured_state_->dst()->SetTensor(std::move(outputs[0]), recv_device); in StartRecv()
|