Searched defs:NestedTensor (Results 1 – 4 of 4) sorted by relevance
48 class NestedTensor(torch.Tensor): class325 def forward(ctx, x: NestedTensor): # type: ignore[override]367 def backward(ctx, gO: NestedTensor): # type: ignore[override]
132 func, a: NestedTensor, a_name: str, b: NestedTensor, b_name: str
221 NestedTensor, enumerator
95 NestedTensor = auto() variable in DispatchKey