/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()
|