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