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