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