MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / MutableRegistry

Function MutableRegistry

tensorflow/core/common_runtime/copy_tensor.cc:44–48  ·  view source on GitHub ↗

We use a vector instead of a map since we expect there to be very few registrations.

Source from the content-addressed store, hash-verified

42// We use a vector instead of a map since we expect there to be very
43// few registrations.
44std::vector<RegistrationInfo>* MutableRegistry() {
45 static std::vector<RegistrationInfo>* registry =
46 new std::vector<RegistrationInfo>;
47 return registry;
48}
49
50void CopyHostToDevice(const Tensor* input, Allocator* cpu_allocator,
51 Allocator* out_allocator, StringPiece edge_name,

Callers 2

ViaDMAMethod · 0.85
RegisterMethod · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected