Searched defs:batch_dim_indices (Results 1 – 2 of 2) sorted by relevance
219 std::vector<int64> batch_dim_indices(num_batch_dims); in QrBlock() local323 std::vector<int64> batch_dim_indices(batch_dims.size()); in CompactWYRepresentation() local
6930 llvm::SmallVector<int64_t, 4> batch_dim_indices(batch_dims.size()); in QRBlock() local7100 llvm::SmallVector<int64_t, 4> batch_dim_indices(batch_dims.size()); in ComputeWYRepresentation() local