Searched refs:batch_group_count_ (Results 1 – 2 of 2) sorted by relevance
2140 batch_group_count_(batch_group_count), in HloConvolutionInstruction()2171 proto.set_batch_group_count(batch_group_count_); in ToProto()2188 if (batch_group_count_ != 1) { in ExtraAttributesToStringImpl()2189 extra.push_back(StrCat("batch_group_count=", batch_group_count_)); in ExtraAttributesToStringImpl()2208 if (batch_group_count_ != other.batch_group_count()) { in IdenticalSlowPath()2226 batch_group_count_, window(), convolution_dimension_numbers_, in CloneWithNewOperandsImpl()2344 batch_group_count_(1), in HloCustomCallInstruction()2361 batch_group_count_(1), in HloCustomCallInstruction()2379 batch_group_count_(1), in HloCustomCallInstruction()2399 batch_group_count_(1), in HloCustomCallInstruction()[all …]
1247 int64 batch_group_count() const { return batch_group_count_; } in batch_group_count()1249 batch_group_count_ = num_batch_groups; in set_batch_group_count()1281 int64 batch_group_count_; variable1450 batch_group_count_ = batch_group_count; in set_batch_group_count()1458 int64 batch_group_count() const { return batch_group_count_; } in batch_group_count()1525 int64 batch_group_count_; variable