Searched refs:TraceDictStrTensor (Results 1 – 1 of 1) sorted by relevance
1224 class TraceDictStrTensor(torch.nn.Module): class1234 torch.jit.trace(TraceDictStrTensor(), x)1236 traced_dict_str_mod = torch.jit.trace(TraceDictStrTensor(), x, strict=False)