Home
last modified time | relevance | path

Searched refs:tensor_wrapper (Results 1 – 13 of 13) sorted by relevance

/aosp_15_r20/external/executorch/backends/qualcomm/runtime/
H A DQnnManager.cpp154 const std::shared_ptr<TensorWrapper>& tensor_wrapper) { in RegisterMem() argument
170 return RegisterCustomMem(data_ptr, custom_mem_base, tensor_wrapper); in RegisterMem()
172 return RegisterIonMem(data_ptr, tensor_wrapper); in RegisterMem()
177 const std::shared_ptr<TensorWrapper>& tensor_wrapper) { in RegisterIonMem() argument
186 tensor_wrapper->GetMemHandle(), data_ptr)) { in RegisterIonMem()
190 tensor_wrapper->GetName().c_str()); in RegisterIonMem()
198 tensor_wrapper->GetName().c_str()); in RegisterIonMem()
203 tensor_wrapper, mem_fd, data_ptr) == Error::Ok, in RegisterIonMem()
213 const std::shared_ptr<TensorWrapper>& tensor_wrapper) { in RegisterCustomMem() argument
215 tensor_wrapper->GetMemHandle(), data_ptr)) { in RegisterCustomMem()
[all …]
H A DQnnManager.h81 const std::shared_ptr<TensorWrapper>& tensor_wrapper);
130 const std::shared_ptr<TensorWrapper>& tensor_wrapper);
134 const std::shared_ptr<TensorWrapper>& tensor_wrapper);
/aosp_15_r20/external/executorch/backends/qualcomm/runtime/backends/
H A DQnnMemManager.cpp25 const std::shared_ptr<TensorWrapper>& tensor_wrapper, in RegisterIonMem() argument
30 {tensor_wrapper->GetRank(), tensor_wrapper->GetDims(), nullptr}, in RegisterIonMem()
31 tensor_wrapper->GetDataType(), in RegisterIonMem()
44 tensor_wrapper->GetName().c_str(), in RegisterIonMem()
48 tensor_wrapper->SetMemHandle(handle); in RegisterIonMem()
52 tensor_wrapper->GetName().c_str()); in RegisterIonMem()
57 const std::shared_ptr<TensorWrapper>& tensor_wrapper, in RegisterCustomMem() argument
65 {tensor_wrapper->GetRank(), tensor_wrapper->GetDims(), nullptr}, in RegisterCustomMem()
66 tensor_wrapper->GetDataType(), in RegisterCustomMem()
89 tensor_wrapper->GetName().c_str(), in RegisterCustomMem()
[all …]
H A DQnnGraphCommon.cpp94 const std::shared_ptr<TensorWrapper>& tensor_wrapper) { in EnsureTensorInQnnGraph() argument
98 if (!tensor_wrapper->IsTensorCreated()) { in EnsureTensorInQnnGraph()
99 Qnn_Tensor_t tensor = tensor_wrapper->CloneTensorStruct(); in EnsureTensorInQnnGraph()
106 const std::string& old_name = tensor_wrapper->GetName(); in EnsureTensorInQnnGraph()
110 tensor_wrapper->SetName(new_name); in EnsureTensorInQnnGraph()
123 tensor_wrapper->UpdateQnnTensorMeta(tensor); in EnsureTensorInQnnGraph()
124 tensor_wrapper->SetTensorCreated(); in EnsureTensorInQnnGraph()
H A DQnnMemManager.h31 const std::shared_ptr<TensorWrapper>& tensor_wrapper,
36 const std::shared_ptr<TensorWrapper>& tensor_wrapper,
57 const std::shared_ptr<TensorWrapper>& tensor_wrapper,
H A DQnnGraphCommon.h52 const std::shared_ptr<TensorWrapper>& tensor_wrapper);
/aosp_15_r20/external/executorch/backends/qualcomm/aot/wrappers/
H A DOpWrapper.cpp20 for (const auto& tensor_wrapper : input_tensors_) { in GetOpConfig() local
21 input_tensor_structs_.emplace_back(tensor_wrapper->CloneTensorStruct()); in GetOpConfig()
24 for (const auto& tensor_wrapper : output_tensors_) { in GetOpConfig() local
25 output_tensor_structs_.emplace_back(tensor_wrapper->CloneTensorStruct()); in GetOpConfig()
H A DOpWrapper.h71 std::shared_ptr<TensorWrapper> tensor_wrapper = CreateTensorWrapper( variable
81 std::make_unique<TensorParamWrapper>(name, tensor_wrapper));
/aosp_15_r20/external/executorch/backends/qualcomm/builders/
H A Dnode_visitor.py266 tensor_wrapper = PyQnnWrapper.TensorWrapper(
280 nodes_to_wrappers[node_name][wrapper_idx] = tensor_wrapper
281 return tensor_wrapper
322 tensor_wrapper = PyQnnWrapper.TensorWrapper(
340 tensor_wrapper = PyQnnWrapper.TensorWrapper(
351 nodes_to_wrappers[node_name][wrapper_idx] = tensor_wrapper
352 return tensor_wrapper
/aosp_15_r20/external/executorch/backends/qualcomm/aot/python/
H A DPyQnnManagerAdaptor.h181 for (const auto& tensor_wrapper : op_wrapper->GetInputTensors()) { in Compile() local
182 set_tensor(tensor_wrapper, inputs); in Compile()
185 for (const auto& tensor_wrapper : op_wrapper->GetOutputTensors()) { in Compile() local
186 set_tensor(tensor_wrapper, outputs); in Compile()
/aosp_15_r20/external/executorch/runtime/core/test/
H A Devalue_test.cpp262 TensorWrapper tensor_wrapper(tf.ones({4, 5})); in TEST_F() local
264 EValue evalue(tensor_wrapper); in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/c/kernels/
H A Dtensor_shape_utils_test.cc41 TF_TensorWrapper tensor_wrapper = TF_TensorWrapper(tf_tensor); in TestShapeMatch() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dcontext.cc256 TF_Tensor* tensor_wrapper = in CreateTensor() local
261 std::swap(result, tensor_wrapper->tensor); in CreateTensor()
262 TF_DeleteTensor(tensor_wrapper); in CreateTensor()