Searched defs:nnz_per_batch (Results 1 – 3 of 3) sorted by relevance
183 auto nnz_per_batch = nnz_per_batch_host.tensor().vec<int32>(); in ComputeAsync() local
229 auto nnz_per_batch = nnz_per_batch_host.tensor().vec<int32>(); in ComputeAsync() local
55 TTypes<int32>::Vec nnz_per_batch) { in operator ()()