| /aosp_15_r20/external/executorch/backends/qualcomm/runtime/ |
| H A D | QnnManager.cpp | 154 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 D | QnnManager.h | 81 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 D | QnnMemManager.cpp | 25 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 D | QnnGraphCommon.cpp | 94 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 D | QnnMemManager.h | 31 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 D | QnnGraphCommon.h | 52 const std::shared_ptr<TensorWrapper>& tensor_wrapper);
|
| /aosp_15_r20/external/executorch/backends/qualcomm/aot/wrappers/ |
| H A D | OpWrapper.cpp | 20 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 D | OpWrapper.h | 71 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 D | node_visitor.py | 266 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 D | PyQnnManagerAdaptor.h | 181 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 D | evalue_test.cpp | 262 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 D | tensor_shape_utils_test.cc | 41 TF_TensorWrapper tensor_wrapper = TF_TensorWrapper(tf_tensor); in TestShapeMatch() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/ |
| H A D | context.cc | 256 TF_Tensor* tensor_wrapper = in CreateTensor() local
261 std::swap(result, tensor_wrapper->tensor); in CreateTensor()
262 TF_DeleteTensor(tensor_wrapper); in CreateTensor()
|