Searched refs:output_batch_size_ (Results 1 – 2 of 2) sorted by relevance
52 output_batch_size_ = output_batch_shape_.num_elements(); in MatMulBCast()54 std::min(x_batch_size_, y_batch_size_) != output_batch_size_; in MatMulBCast()57 ComputeBatchIndices(output_batch_size_, batch_bcast_->x_reshape(), in MatMulBCast()59 ComputeBatchIndices(output_batch_size_, batch_bcast_->y_reshape(), in MatMulBCast()69 const int64 output_batch_size() const { return output_batch_size_; } in output_batch_size()93 int64 output_batch_size_ = 1; variable
99 const int64 output_batch_size() const { return output_batch_size_; } in output_batch_size()120 int64 output_batch_size_; variable143 output_batch_size_ = 1; in BCastList()166 output_batch_size_ = elements; in BCastList()227 output_batch_size_ = mul_dims(output_batch_size_, output_.back()); in BCastList()305 output_batch_size_ > 0) { in BCastList()307 ComputeBatchIndices(output_batch_size_, reshape_[i], bcast_[i], in BCastList()