Home
last modified time | relevance | path

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

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