Searched refs:ViewNestedFromBuffer (Results 1 – 3 of 3) sorted by relevance
18 from .nested_tensor import buffer_from_jagged, NestedTensor, ViewNestedFromBuffer698 attention = ViewNestedFromBuffer.apply(737 return ViewNestedFromBuffer.apply(769 attn_out = ViewNestedFromBuffer.apply(attn_out, offsets)
343 class ViewNestedFromBuffer(torch.autograd.Function): class
23 ViewNestedFromBuffer,247 nt = ViewNestedFromBuffer.apply(259 nt = ViewNestedFromBuffer.apply(values, offsets, metadata_cache)