Searched refs:batch_dim_ids (Results 1 – 1 of 1) sorted by relevance
81 std::vector<int64> batch_dim_ids(batch_dims.size()); in House() local82 std::iota(batch_dim_ids.begin(), batch_dim_ids.end(), 0); in House()115 *v = e_k + Div(x_after_k, divisor, /*broadcast_dimensions=*/batch_dim_ids); in House()