Searched refs:batch_group_count_ (Results 1 – 2 of 2) sorted by relevance
1861 batch_group_count_(batch_group_count), in HloConvolutionInstruction()1892 proto.set_batch_group_count(batch_group_count_); in ToProto()1909 if (batch_group_count_ != 1) { in ExtraAttributesToStringImpl()1910 extra.push_back(StrCat("batch_group_count=", batch_group_count_)); in ExtraAttributesToStringImpl()1930 if (batch_group_count_ != other.batch_group_count()) { in IdenticalSlowPath()1948 batch_group_count_, window(), convolution_dimension_numbers_, in CloneWithNewOperandsImpl()2051 batch_group_count_(1), in HloCustomCallInstruction()2066 batch_group_count_(1), in HloCustomCallInstruction()2087 proto.set_batch_group_count(batch_group_count_); in ToProto()2111 if (batch_group_count_ != 1) { in ExtraAttributesToStringImpl()[all …]
1051 int64 batch_group_count() const { return batch_group_count_; } in batch_group_count()1083 int64 batch_group_count_; variable1207 batch_group_count_ = batch_group_count; in set_batch_group_count()1210 int64 batch_group_count() const { return batch_group_count_; } in batch_group_count()1245 int64 batch_group_count_; variable