Searched defs:output_shards_chunks (Results 1 – 2 of 2) sorted by relevance
378 auto output_shards_chunks = MatrixRef<Tensor>(output_shards, num_batches); in batchedTensorForLoopFallback() local502 auto output_shards_chunks = MatrixRef<Tensor>(output_shards, num_components); in batchedNestedTensorForLoopFallback() local
358 auto output_shards_chunks = MatrixRef<Tensor>(output_shards, num_batches); in batchedTensorForLoopFallback() local