Home
last modified time | relevance | path

Searched refs:import_memory (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_matmul_acl.cc147 acl_obj.lhs_acc_tensor.allocator()->import_memory(lhs); in MatMulF32()
149 acl_obj.rhs_tensor.allocator()->import_memory(rhs); in MatMulF32()
152 acl_obj.rhs_acc_tensor.allocator()->import_memory(rhs); in MatMulF32()
154 acl_obj.lhs_tensor.allocator()->import_memory(lhs); in MatMulF32()
157 acl_obj.lhs_acc_tensor.allocator()->import_memory(lhs); in MatMulF32()
159 acl_obj.rhs_acc_tensor.allocator()->import_memory(rhs); in MatMulF32()
163 acl_obj.lhs_tensor.allocator()->import_memory(lhs); in MatMulF32()
164 acl_obj.rhs_tensor.allocator()->import_memory(rhs); in MatMulF32()
167 acl_obj.out_tensor.allocator()->import_memory(out); in MatMulF32()
Druntime_conv2d_acl.cc114 acl_conv_obj.input_tensor.allocator()->import_memory(lhs); in ACLDepthwiseConvImpl()
115 acl_conv_obj.kernel_tensor.allocator()->import_memory(rhs); in ACLDepthwiseConvImpl()
116 acl_conv_obj.output_tensor.allocator()->import_memory(out); in ACLDepthwiseConvImpl()
219 acl_conv_obj.input_tensor.allocator()->import_memory(lhs); in ACLGemmConvImpl()
220 acl_conv_obj.kernel_tensor.allocator()->import_memory(rhs); in ACLGemmConvImpl()
221 acl_conv_obj.output_tensor.allocator()->import_memory(out); in ACLGemmConvImpl()
/external/ComputeLibrary/tests/validation/NEON/UNIT/
DTensorAllocator.cpp64 ARM_COMPUTE_ASSERT(!bool(t1.allocator()->import_memory(nullptr))); in TEST_SUITE()
71 ARM_COMPUTE_ASSERT(!bool(t2.allocator()->import_memory(data.get()))); in TEST_SUITE()
78 ARM_COMPUTE_ASSERT(!bool(t3.allocator()->import_memory(data.get()))); in TEST_SUITE()
84 ARM_COMPUTE_ASSERT(bool(t4.allocator()->import_memory(data.get()))); in TEST_SUITE()
117 ARM_COMPUTE_ASSERT(bool(tensor.allocator()->import_memory(aligned_ptr))); in TEST_CASE()
163 ARM_COMPUTE_ASSERT(bool(tensor.allocator()->import_memory(raw_data.get()))); in TEST_CASE()
228 ARM_COMPUTE_ASSERT(bool(tensor.allocator()->import_memory(data))); in TEST_CASE()
/external/ComputeLibrary/tests/validation/CL/UNIT/
DTensorAllocator.cpp183 ARM_COMPUTE_ASSERT(!bool(t1.allocator()->import_memory(cl::Buffer()))); in TEST_CASE()
190 ARM_COMPUTE_ASSERT(!bool(t2.allocator()->import_memory(buf))); in TEST_CASE()
197 ARM_COMPUTE_ASSERT(!bool(t3.allocator()->import_memory(buf))); in TEST_CASE()
203 ARM_COMPUTE_ASSERT(bool(t4.allocator()->import_memory(buf))); in TEST_CASE()
245 ARM_COMPUTE_ASSERT(bool(tensor.allocator()->import_memory(wrapped_buffer))); in TEST_CASE()
313 ARM_COMPUTE_ASSERT(bool(tensor.allocator()->import_memory(wrapped_buffer))); in TEST_CASE()
/external/ComputeLibrary/src/gpu/cl/utils/
DClAuxTensorHandler.h69 _tensor.allocator()->import_memory(packed_tensor->cl_buffer()); in _tensor()
79 _tensor.allocator()->import_memory(tensor.cl_buffer()); in CLAuxTensorHandler()
/external/ComputeLibrary/src/cpu/utils/
DCpuAuxTensorHandler.h69 _tensor.allocator()->import_memory(packed_tensor->buffer()); in _tensor()
79 _tensor.allocator()->import_memory(tensor.buffer()); in CpuAuxTensorHandler()
/external/armnn/src/backends/cl/
DClImportTensorHandle.hpp142 … m_Tensor.allocator()->import_memory(cl::Buffer(static_cast<cl_mem>(memory))); in Import()
163 … m_Tensor.allocator()->import_memory(cl::Buffer(static_cast<cl_mem>(memory))); in Import()
238 arm_compute::Status status = m_Tensor.allocator()->import_memory(wrappedBuffer); in ClImport()
/external/ComputeLibrary/arm_compute/runtime/
DTensorAllocator.h107 Status import_memory(void *memory);
/external/crosvm/rutabaga_gfx/src/rutabaga_gralloc/vulkano_gralloc/sys/
Dwindows.rs38 pub(crate) unsafe fn import_memory( in import_memory() method
Dlinux.rs42 pub(crate) unsafe fn import_memory( in import_memory() method
/external/ComputeLibrary/src/cpu/
DCpuTensor.cpp71 const auto st = _legacy_tensor->allocator()->import_memory(handle); in import()
/external/ComputeLibrary/arm_compute/runtime/CL/
DCLTensorAllocator.h124 Status import_memory(cl::Buffer buffer);
/external/armnn/src/backends/neon/
DNeonTensorHandle.hpp140 arm_compute::Status status = m_Tensor.allocator()->import_memory(memory); in Import()
161 arm_compute::Status status = m_Tensor.allocator()->import_memory(memory); in Import()
/external/ComputeLibrary/src/runtime/
DTensorAllocator.cpp153 Status TensorAllocator::import_memory(void *memory) in import_memory() function in TensorAllocator
/external/ComputeLibrary/src/runtime/CL/
DCLTensorAllocator.cpp167 Status CLTensorAllocator::import_memory(cl::Buffer buffer) in import_memory() function in arm_compute::CLTensorAllocator
/external/ComputeLibrary/src/runtime/NEON/functions/
DNEFFTConvolutionLayer.cpp330 _reshaped_output.allocator()->import_memory(_itransformed_output.buffer()); in run()
/external/ComputeLibrary/src/runtime/CL/functions/
DCLFFTConvolutionLayer.cpp336 _reshaped_output.allocator()->import_memory(_itransformed_output.cl_buffer()); in run()
/external/crosvm/rutabaga_gfx/src/rutabaga_gralloc/
Dvulkano_gralloc.rs509 VulkanoGralloc::import_memory( in import_and_map()
/external/ComputeLibrary/src/cpu/operators/
DCpuGemmConv2d.cpp580 gemm3d.allocator()->import_memory(out_to_use->buffer()); in run()
/external/ComputeLibrary/docs/user_guide/
Dlibrary.dox340 tensor.allocator()->import_memory(external_ptr);