Searched refs:output_batch_size_ (Results 1 – 3 of 3) sorted by relevance
30 output_batch_size_ = output_shape_.num_elements(); in MatMulBCast()32 std::min(x_batch_size_, y_batch_size_) != output_batch_size_; in MatMulBCast()35 ComputeBatchIndices(output_batch_size_, batch_bcast_->x_reshape(), in MatMulBCast()37 ComputeBatchIndices(output_batch_size_, batch_bcast_->y_reshape(), in MatMulBCast()
99 const int64 output_batch_size() const { return output_batch_size_; } in output_batch_size()120 int64 output_batch_size_; variable137 output_batch_size_ = 1; in BCastList()160 output_batch_size_ = elements; in BCastList()221 output_batch_size_ *= output_.back(); in BCastList()298 output_batch_size_ > 0) { in BCastList()300 ComputeBatchIndices(output_batch_size_, reshape_[i], bcast_[i], in BCastList()
39 const int64 output_batch_size() const { return output_batch_size_; } in output_batch_size()63 int64 output_batch_size_; variable