Searched refs:set_batch_group_count (Results 1 – 6 of 6) sorted by relevance
1892 proto.set_batch_group_count(batch_group_count_); in ToProto()2087 proto.set_batch_group_count(batch_group_count_); in ToProto()2192 cloned->set_batch_group_count(batch_group_count_); in CloneWithNewOperandsImpl()
474 ->set_batch_group_count( in CreateFromProto()3479 void HloInstruction::set_batch_group_count(int64 batch_group_count) { in set_batch_group_count() function in xla::HloInstruction3480 Cast<HloCustomCallInstruction>(this)->set_batch_group_count( in set_batch_group_count()
1566 void set_batch_group_count(int64 batch_group_count);
1206 void set_batch_group_count(int64 batch_group_count) { in set_batch_group_count() function
1557 instruction->set_batch_group_count(*batch_group_count); in ParseInstructionRhs()
1199 instr.set_batch_group_count(batch_group_count); in ConvGeneralDilated()