Searched defs:BatchedNestedTensor (Results 1 – 3 of 3) sorted by relevance
758 TORCH_LIBRARY_IMPL(_, BatchedNestedTensor, m) { in TORCH_LIBRARY_IMPL() argument763 TORCH_LIBRARY_IMPL(aten, BatchedNestedTensor, m) { in TORCH_LIBRARY_IMPL() argument
374 BatchedNestedTensor, enumerator
119 BatchedNestedTensor = auto() variable in DispatchKey