Home
last modified time | relevance | path

Searched defs:create_tensor (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/
H A DClContext.cpp92 ITensorV2 *ClContext::create_tensor(const AclTensorDescriptor &desc, bool allocate) in create_tensor() function in arm_compute::gpu::opencl::ClContext
/aosp_15_r20/external/ComputeLibrary/src/graph/backends/NEON/
H A DNEDeviceBackend.cpp116 std::unique_ptr<ITensorHandle> NEDeviceBackend::create_tensor(const Tensor &tensor) in create_tensor() function in arm_compute::graph::backends::NEDeviceBackend
/aosp_15_r20/external/ComputeLibrary/src/graph/backends/CL/
H A DCLDeviceBackend.cpp164 std::unique_ptr<ITensorHandle> CLDeviceBackend::create_tensor(const Tensor &tensor) in create_tensor() function in arm_compute::graph::backends::CLDeviceBackend
/aosp_15_r20/external/ComputeLibrary/src/cpu/
H A DCpuContext.cpp171 ITensorV2 *CpuContext::create_tensor(const AclTensorDescriptor &desc, bool allocate) in create_tensor() function in arm_compute::cpu::CpuContext
/aosp_15_r20/external/ComputeLibrary/src/graph/
H A DGraph.cpp155 TensorID Graph::create_tensor(const TensorDescriptor &desc) in create_tensor() function in arm_compute::graph::Graph
/aosp_15_r20/external/pytorch/torch/cuda/
H A D_sanitizer.py188 def create_tensor( member in _TensorsAccessed
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/
H A Ddist_autograd_test.py79 def create_tensor(): function