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