Home
last modified time | relevance | path

Searched refs:external_tensor (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/c/
H A DAclTensor.cpp68 extern "C" AclStatus AclCreateTensor(AclTensor *external_tensor, in AclCreateTensor() argument
92 *external_tensor = tensor; in AclCreateTensor()
97 extern "C" AclStatus AclMapTensor(AclTensor external_tensor, void **handle) in AclMapTensor() argument
101 auto tensor = get_internal(external_tensor); in AclMapTensor()
116 extern "C" AclStatus AclUnmapTensor(AclTensor external_tensor, void *handle) in AclUnmapTensor() argument
122 auto tensor = get_internal(external_tensor); in AclUnmapTensor()
130 extern "C" AclStatus AclTensorImport(AclTensor external_tensor, void *handle, AclImportMemoryType t… in AclTensorImport() argument
134 auto tensor = get_internal(external_tensor); in AclTensorImport()
144 extern "C" AclStatus AclDestroyTensor(AclTensor external_tensor) in AclDestroyTensor() argument
148 auto tensor = get_internal(external_tensor); in AclDestroyTensor()
H A DAclTensorPack.cpp32 StatusCode PackTensorInternal(TensorPack &pack, AclTensor external_tensor, int32_t slot_id) in PackTensorInternal() argument
35 auto tensor = get_internal(external_tensor); in PackTensorInternal()
70 extern "C" AclStatus AclPackTensor(AclTensorPack external_pack, AclTensor external_tensor, int32_t … in AclPackTensor() argument
76 ARM_COMPUTE_RETURN_CENUM_ON_FAILURE(PackTensorInternal(*pack, external_tensor, slot_id)); in AclPackTensor()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
H A Dinference_context.cc155 for (const auto& external_tensor : create_info.external_immutable_tensors) { in InitFromGraph() local
157 dynamic_cast<MetalSpatialTensor*>(external_tensor.second); in InitFromGraph()
161 external_immutable_tensors_[external_tensor.first] = metal_spatial_tensor; in InitFromGraph()
164 for (const auto& external_tensor : create_info.external_mutable_tensors) { in InitFromGraph() local
166 CreateTensor(device_id, tensors_descs_[external_tensor.first], in InitFromGraph()
167 &temp_external_tensors[external_tensor.first])); in InitFromGraph()
168 external_mutable_tensors_[external_tensor.first] = in InitFromGraph()
169 &temp_external_tensors[external_tensor.first]; in InitFromGraph()
178 for (auto& external_tensor : external_mutable_tensors_) { in InitFromGraph() local
179 external_tensor.second = nullptr; in InitFromGraph()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
H A Dinference_context.cc264 for (const auto& external_tensor : create_info.external_immutable_tensors) { in InitFromGpuModel() local
265 auto* cl_spatial_tensor = dynamic_cast<Tensor*>(external_tensor.second); in InitFromGpuModel()
269 external_immutable_tensors_[external_tensor.first] = cl_spatial_tensor; in InitFromGpuModel()
272 for (const auto& external_tensor : create_info.external_mutable_tensors) { in InitFromGpuModel() local
275 gpu_model->tensors[external_tensor.first], in InitFromGpuModel()
276 &temp_external_tensors[external_tensor.first])); in InitFromGpuModel()
277 external_mutable_tensors_[external_tensor.first] = in InitFromGpuModel()
278 &temp_external_tensors[external_tensor.first]; in InitFromGpuModel()
305 for (auto& external_tensor : external_mutable_tensors_) { in InitFromGpuModel() local
306 external_tensor.second = nullptr; in InitFromGpuModel()
[all …]
/aosp_15_r20/external/ComputeLibrary/src/c/cl/
H A DAclOpenClExt.cpp163 extern "C" AclStatus AclGetClMem(AclTensor external_tensor, cl_mem *opencl_mem) in AclGetClMem() argument
166 ITensorV2 *tensor = get_internal(external_tensor); in AclGetClMem()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/
H A Dmetal_delegate.mm464 for (auto& external_tensor : in_out_tensors_) {
466 inference_context_.SetTensor(external_tensor.first, external_tensor.second.get()));