MCPcopy Create free account
hub / github.com/pytorch/pytorch / get_gpu_blob_name

Method get_gpu_blob_name

caffe2/python/optimizer.py:74–82  ·  view source on GitHub ↗
(self, base_str, gpu_id, node_name)

Source from the content-addressed store, hash-verified

72 return "%s_%d_%s%s_cpu" % (classname, self._instance_num, base_str, node_name)
73
74 def get_gpu_blob_name(self, base_str, gpu_id, node_name):
75 classname = self.__class__.__name__
76 return "%s_%d_%s%s_gpu%d" % (
77 classname,
78 self._instance_num,
79 base_str,
80 node_name,
81 gpu_id,
82 )
83
84 @property
85 def attributes(self):

Callers 2

make_unique_blob_nameMethod · 0.95
GetLearningRateBlobNamesFunction · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected