Searched refs:x_batch_indices_ (Results 1 – 2 of 2) sorted by relevance
50 const std::vector<int64>& x_batch_indices() const { return x_batch_indices_; } in x_batch_indices()64 std::vector<int64> x_batch_indices_; variable
36 batch_bcast_->x_bcast(), &x_batch_indices_); in MatMulBCast()