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