Searched refs:input_nestedtensor (Results 1 – 1 of 1) sorted by relevance
171 const at::Tensor& input_nestedtensor, in view_as_dense() argument177 const auto* tensor_impl = get_nested_tensor_impl(input_nestedtensor); in view_as_dense()191 {Nnz, input_nestedtensor.size(1), num_heads, head_dim}, in view_as_dense()