Searched defs:pad_tensor (Results 1 – 2 of 2) sorted by relevance
170 def pad_tensor(tensor: torch.Tensor, pad_dim: int, pad_size: int) -> torch.Tensor: function
835 auto& pad_tensor = op_sig.inputs.at(1); in CheckGpuDelegateCompatibility() local