Searched refs:nestedtensor (Results 1 – 2 of 2) sorted by relevance
16 int64_t get_nnz(const Tensor& nestedtensor) { in get_nnz() argument17 auto* nt_impl = get_nested_tensor_impl(nestedtensor); in get_nnz()20 const int64_t batch_size = nestedtensor.size(0); in get_nnz()
99 * nestedtensor