Lines Matching refs:transform_tensor

7552     transform_tensor = torch.from_numpy(rotation_ary).to(device, torch.float32)
7553     transform_tensor = transform_tensor[:2].unsqueeze(0)
7555     return transform_tensor, transform_ary, grid_ary
7625     transform_tensor = torch.from_numpy(torchRotation_ary).to(device, torch.float32)
7626     transform_tensor = transform_tensor[:3].unsqueeze(0)
7628     return transform_tensor, transform_ary, grid_ary
7935         transform_tensor, transform_ary, offset = \
7948             transform_tensor,
7978             transform_tensor, transform_ary, offset = \
7998                 transform_tensor,
8028         transform_tensor, transform_ary, offset = \
8041             transform_tensor,
8111             transform_tensor, transform_ary, grid_ary = \
8123                 transform_tensor,
8167             transform_tensor, transform_ary, grid_ary = \
8179                 transform_tensor,