Searched refs:y_batch_size_ (Results 1 – 1 of 1) sorted by relevance
50 y_batch_size_ = TensorShape(batch_bcast_->y_reshape()).num_elements(); in MatMulBCast()54 std::min(x_batch_size_, y_batch_size_) != output_batch_size_; in MatMulBCast()71 const int64 y_batch_size() const { return y_batch_size_; } in y_batch_size()91 int64 y_batch_size_ = 1; variable