MCPcopy Create free account
hub / github.com/ARM-software/ComputeLibrary / configure_all_tensors

Function configure_all_tensors

src/graph/detail/ExecutionHelpers.cpp:56–71  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

54}
55
56void configure_all_tensors(Graph &g)
57{
58 auto &tensors = g.tensors();
59
60 for (auto &tensor : tensors)
61 {
62 if (tensor && tensor->handle() == nullptr)
63 {
64 Target target = tensor->desc().target;
65 backends::IDeviceBackend &backend = backends::BackendRegistry::get().get_backend(target);
66 std::unique_ptr<ITensorHandle> handle = backend.create_tensor(*tensor);
67 ARM_COMPUTE_ERROR_ON_MSG(!handle, "Couldn't create backend handle!");
68 tensor->set_handle(std::move(handle));
69 }
70 }
71}
72
73void allocate_all_input_tensors(INode &node)
74{

Callers 1

finalize_graphMethod · 0.85

Calls 3

handleMethod · 0.80
set_handleMethod · 0.80
create_tensorMethod · 0.45

Tested by

no test coverage detected