Searched refs:y_batch_indices_ (Results 1 – 2 of 2) sorted by relevance
55 const std::vector<int64>& y_batch_indices() const { return y_batch_indices_; } in y_batch_indices()65 std::vector<int64> y_batch_indices_; variable
38 batch_bcast_->y_bcast(), &y_batch_indices_); in MatMulBCast()