Searched refs:torch_tensor (Results 1 – 4 of 4) sorted by relevance
141 def to_tvm_tensor(torch_tensor): argument143 if torch_tensor.dtype == torch.bool:146 return tvm.nd.array(torch_tensor.cpu().numpy())147 return tvm.nd.from_dlpack(torch_tensor)
32 torch_tensor: torch.Tensor,40 flatbuffer_tensor.scalar_type, scalar_type_enum(torch_tensor.dtype)44 self.assertEqual(flatbuffer_tensor.sizes, list(torch_tensor.size()))45 self.assertEqual(flatbuffer_tensor.requires_grad, torch_tensor.requires_grad)
868 torch_tensor = "torch.Tensor."870 if n.startswith(torch_tensor):871 names_sanitized.append(n[len(torch_tensor) :])
895 for name, torch_tensor in itertools.chain(904 if not isinstance(torch_tensor, torch.Tensor):909 ir_tensor = TorchTensor(torch_tensor, name=name)913 torch_tensor,