Searched refs:x_batch_size_ (Results 1 – 2 of 2) sorted by relevance
40 const int64 x_batch_size() const { return x_batch_size_; } in x_batch_size()60 int64 x_batch_size_; variable
27 x_batch_size_ = TensorShape(batch_bcast_->x_reshape()).num_elements(); in MatMulBCast()32 std::min(x_batch_size_, y_batch_size_) != output_batch_size_; in MatMulBCast()