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

Function MakeResourceHandle

tensorflow/core/framework/resource_mgr.cc:35–51  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

33static std::atomic<int64> current_id_;
34
35ResourceHandle MakeResourceHandle(
36 const string& container, const string& name, const DeviceBase& device,
37 const TypeIndex& type_index,
38 const std::vector<DtypeAndPartialTensorShape>& dtypes_and_shapes) {
39 ResourceHandle result;
40 result.set_device(device.physical_name());
41 result.set_container(container);
42 if (name == ResourceHandle::ANONYMOUS_NAME) {
43 result.set_name(strings::StrCat("_AnonymousVar", current_id_.fetch_add(1)));
44 } else {
45 result.set_name(name);
46 }
47 result.set_hash_code(type_index.hash_code());
48 result.set_maybe_type_name(type_index.name());
49 result.set_dtypes_and_shapes(dtypes_and_shapes);
50 return result;
51}
52
53Status MakeResourceHandleToOutput(OpKernelContext* context, int output_index,
54 const string& container, const string& name,

Callers 1

Calls 9

set_containerMethod · 0.80
set_hash_codeMethod · 0.80
set_maybe_type_nameMethod · 0.80
set_dtypes_and_shapesMethod · 0.80
nameMethod · 0.65
StrCatFunction · 0.50
set_deviceMethod · 0.45
set_nameMethod · 0.45
hash_codeMethod · 0.45

Tested by

no test coverage detected