Home
last modified time | relevance | path

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

/external/pytorch/test/distributed/pipelining/
Dmodel_registry.py33 DEFAULT_DHID = 512 variable in ModelWithKwargs
43 def forward(self, x, y=torch.zeros(DEFAULT_BATCH_SIZE, DEFAULT_DHID)): argument