| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | TensorWrapper.cpp | 41 void TensorWrapper::refreshMetadata() { in refreshMetadata()
56 static c10::intrusive_ptr<TensorWrapper> makeTensorWrapperPtr(const Tensor& tensor, int64_t level, … in makeTensorWrapperPtr()
61 return c10::make_intrusive<TensorWrapper>(key_set, tensor, level, life_handle); in makeTensorWrapperPtr()
81 auto result = at::detail::make_tensor<TensorWrapper>( in unsafeMakeTensorWrapper()
110 bool TensorWrapper::is_alive() const { in is_alive()
114 c10::intrusive_ptr<TensorImpl> TensorWrapper::shallow_copy_and_detach( in shallow_copy_and_detach()
125 c10::intrusive_ptr<TensorImpl> TensorWrapper::shallow_copy_and_detach( in shallow_copy_and_detach()
136 void TensorWrapper::shallow_copy_from(const c10::intrusive_ptr<TensorImpl>& impl) { in shallow_copy_from()
140 TensorWrapper::TensorWrapper( in TensorWrapper() function in at::functorch::TensorWrapper
162 const char* TensorWrapper::tensorimpl_type_name() const { in tensorimpl_type_name()
[all …]
|
| H A D | TensorWrapper.h | 35 struct TORCH_API TensorWrapper : public c10::TensorImpl { struct
36 explicit TensorWrapper(
99 TORCH_API TensorWrapper* maybeGetTensorWrapper(const Tensor& tensor);
|
| /aosp_15_r20/external/executorch/backends/qualcomm/runtime/ |
| H A D | QnnManager.h | 36 std::vector<std::shared_ptr<TensorWrapper>>& inputs,
37 std::vector<std::shared_ptr<TensorWrapper>>& outputs);
81 const std::shared_ptr<TensorWrapper>& tensor_wrapper);
92 std::vector<std::shared_ptr<TensorWrapper>> GetGraphInputs( in GetGraphInputs()
95 ? std::vector<std::shared_ptr<TensorWrapper>>() in GetGraphInputs()
99 std::vector<std::shared_ptr<TensorWrapper>> GetGraphOutputs( in GetGraphOutputs()
102 ? std::vector<std::shared_ptr<TensorWrapper>>() in GetGraphOutputs()
124 std::unordered_map<std::string, std::vector<std::shared_ptr<TensorWrapper>>>
126 std::unordered_map<std::string, std::vector<std::shared_ptr<TensorWrapper>>>
130 const std::shared_ptr<TensorWrapper>& tensor_wrapper);
[all …]
|
| H A D | QnnManager.cpp | 30 const std::shared_ptr<TensorWrapper>& a, in CompareExportedInput()
31 const std::shared_ptr<TensorWrapper>& b) { in CompareExportedInput()
154 const std::shared_ptr<TensorWrapper>& tensor_wrapper) { in RegisterMem()
177 const std::shared_ptr<TensorWrapper>& tensor_wrapper) { in RegisterIonMem()
213 const std::shared_ptr<TensorWrapper>& tensor_wrapper) { in RegisterCustomMem()
335 std::shared_ptr<TensorWrapper> tensor_wrapper = CreateTensorWrapper(tensor); in AllocateTensor()
346 std::shared_ptr<TensorWrapper> tensor_wrapper = in AllocateTensor()
366 std::vector<std::shared_ptr<TensorWrapper>>& inputs, in AllocateTensor()
367 std::vector<std::shared_ptr<TensorWrapper>>& outputs) { in AllocateTensor()
510 std::vector<std::shared_ptr<TensorWrapper>> graph_inputs, graph_outputs, in CompileQcir()
[all …]
|
| H A D | QnnExecuTorchBackend.cpp | 109 std::vector<std::shared_ptr<TensorWrapper>> input_tensors = in execute()
111 std::vector<std::shared_ptr<TensorWrapper>> output_tensors = in execute()
|
| /aosp_15_r20/external/executorch/backends/qualcomm/aot/wrappers/ |
| H A D | TensorWrapper.cpp | 65 TensorWrapper::TensorWrapper( in TensorWrapper() function in executorch::backends::qnn::TensorWrapper
111 Error TensorWrapper::FillDataBuffer(const void* data, bool copy_data) { in FillDataBuffer()
130 Error TensorWrapper::AllocateDataBuffer() { in AllocateDataBuffer()
143 void TensorWrapper::UpdateQnnTensorMeta(const Qnn_Tensor_t& tensor_src) { in UpdateQnnTensorMeta()
147 Error TensorWrapper::SetName(const std::string& name) { in SetName()
153 Error TensorWrapper::SetMemHandle(Qnn_MemHandle_t mem_handle) { in SetMemHandle()
160 std::shared_ptr<TensorWrapper> CreateTensorWrapper( in CreateTensorWrapper()
174 return std::make_shared<TensorWrapper>( in CreateTensorWrapper()
186 std::shared_ptr<TensorWrapper> CreateTensorWrapper( in CreateTensorWrapper()
208 std::shared_ptr<TensorWrapper> CreateTensorWrapper(const Qnn_Tensor_t& tensor) { in CreateTensorWrapper()
|
| H A D | OpWrapper.h | 52 const std::vector<std::shared_ptr<TensorWrapper>>& tensors) { in AddInputTensors()
57 const std::vector<std::shared_ptr<TensorWrapper>>& tensors) { in AddOutputTensors()
71 std::shared_ptr<TensorWrapper> tensor_wrapper = CreateTensorWrapper(
95 const std::vector<std::shared_ptr<TensorWrapper>>& GetInputTensors() { in GetInputTensors()
99 const std::vector<std::shared_ptr<TensorWrapper>>& GetOutputTensors() { in GetOutputTensors()
111 std::vector<std::shared_ptr<TensorWrapper>> input_tensors_;
112 std::vector<std::shared_ptr<TensorWrapper>> output_tensors_;
|
| H A D | TensorWrapper.h | 23 class TensorWrapper {
25 explicit TensorWrapper(
116 std::shared_ptr<TensorWrapper> CreateTensorWrapper(
128 std::shared_ptr<TensorWrapper> CreateTensorWrapper(
138 std::shared_ptr<TensorWrapper> CreateTensorWrapper(const Qnn_Tensor_t& tensor);
|
| H A D | TensorParamWrapper.h | 23 std::shared_ptr<TensorWrapper> static_tensor) in TensorParamWrapper()
40 std::shared_ptr<TensorWrapper> GetTensorWrapper() { in GetTensorWrapper()
45 std::shared_ptr<TensorWrapper> static_tensor_wrapper_;
|
| H A D | CMakeLists.txt | 10 PUBLIC ${CMAKE_CURRENT_LIST_DIR}/TensorWrapper.cpp
11 ${CMAKE_CURRENT_LIST_DIR}/TensorWrapper.h
|
| /aosp_15_r20/external/executorch/backends/qualcomm/builders/ |
| H A D | node_visitor.py | 260 nodes_to_wrappers: Dict[str, Dict[int, PyQnnWrapper.TensorWrapper]], argument
262 ) -> PyQnnWrapper.TensorWrapper:
266 tensor_wrapper = PyQnnWrapper.TensorWrapper(
288 nodes_to_wrappers: Dict[str, Dict[int, PyQnnWrapper.TensorWrapper]], argument
292 ) -> PyQnnWrapper.TensorWrapper:
322 tensor_wrapper = PyQnnWrapper.TensorWrapper(
340 tensor_wrapper = PyQnnWrapper.TensorWrapper(
357 nodes_to_wrappers: Dict[str, Dict[int, PyQnnWrapper.TensorWrapper]], argument
|
| H A D | op_skip_ops.py | 27 nodes_to_wrappers: Dict[torch.fx.Node, PyQnnWrapper.TensorWrapper], argument
43 nodes_to_wrappers: Dict[torch.fx.Node, PyQnnWrapper.TensorWrapper], argument
|
| H A D | op_conv2d.py | 108 nodes_to_wrappers: Dict[str, PyQnnWrapper.TensorWrapper], argument
268 nodes_to_wrappers: Dict[str, PyQnnWrapper.TensorWrapper], argument
|
| H A D | op_reshape.py | 26 nodes_to_wrappers: Dict[torch.fx.Node, PyQnnWrapper.TensorWrapper], argument
|
| H A D | op_gelu.py | 27 nodes_to_wrappers: Dict[torch.fx.Node, PyQnnWrapper.TensorWrapper], argument
|
| H A D | op_sqrt.py | 26 nodes_to_wrappers: Dict[torch.fx.Node, PyQnnWrapper.TensorWrapper], argument
|
| H A D | op_hardswish.py | 27 nodes_to_wrappers: Dict[torch.fx.Node, PyQnnWrapper.TensorWrapper], argument
|
| H A D | op_ceil.py | 26 nodes_to_wrappers: Dict[torch.fx.Node, PyQnnWrapper.TensorWrapper], argument
|
| H A D | op_tanh.py | 27 nodes_to_wrappers: Dict[torch.fx.Node, PyQnnWrapper.TensorWrapper], argument
|
| /aosp_15_r20/external/executorch/backends/qualcomm/aot/python/ |
| H A D | PyQnnWrapperAdaptor.h | 29 const std::vector<std::shared_ptr<TensorWrapper>>& tensors) { in AddInputTensors()
34 const std::vector<std::shared_ptr<TensorWrapper>>& tensors) { in AddOutputTensors()
103 explicit PyQnnTensorWrapper(const std::shared_ptr<TensorWrapper>& wrapper) { in PyQnnTensorWrapper()
183 std::shared_ptr<TensorWrapper> tensor_wrapper_;
|
| H A D | PyQnnWrapperAdaptor.cpp | 69 std::shared_ptr<TensorWrapper> CreateTensorWrapper( in CreateTensorWrapper()
185 py::class_<TensorWrapper, std::shared_ptr<TensorWrapper>>(m, "TensorWrapper") in PYBIND11_MODULE()
246 .def(py::init<const std::shared_ptr<TensorWrapper>&>()) in PYBIND11_MODULE()
|
| H A D | PyQnnManagerAdaptor.h | 163 auto set_tensor = [&](const std::shared_ptr<TensorWrapper>& wrapper, in Compile()
271 for (const std::shared_ptr<TensorWrapper>& input : in GetGraphInputs()
280 for (const std::shared_ptr<TensorWrapper>& output : in GetGraphOutputs()
|
| /aosp_15_r20/external/executorch/backends/qualcomm/runtime/backends/ |
| 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 | QnnMemManager.cpp | 25 const std::shared_ptr<TensorWrapper>& tensor_wrapper, in RegisterIonMem()
57 const std::shared_ptr<TensorWrapper>& tensor_wrapper, in RegisterCustomMem()
150 const std::shared_ptr<TensorWrapper>& tensor_wrapper, in SetMemHandle()
|
| /aosp_15_r20/external/executorch/runtime/core/test/ |
| H A D | evalue_test.cpp | 37 class TensorWrapper { class
39 explicit TensorWrapper(exec_aten::Tensor tensor) in TensorWrapper() function in TensorWrapper
262 TensorWrapper tensor_wrapper(tf.ones({4, 5})); in TEST_F()
|