Searched defs:batch_dim_indices (Results 1 – 1 of 1) sorted by relevance
173 std::vector<int64> batch_dim_indices(num_batch_dims); in QRBlock() local255 std::vector<int64> batch_dim_indices(batch_dims.size()); in ComputeWYRepresentation() local