Searched refs:output_elements (Results 1 – 2 of 2) sorted by relevance
193 std::vector<HloInstruction*> output_elements(crs->operand_count()); in HandleCrossReplicaSum() local198 output_elements[i] = computation_->AddInstruction( in HandleCrossReplicaSum()205 output_elements[i] = in HandleCrossReplicaSum()210 HloInstruction::CreateTuple(output_elements)); in HandleCrossReplicaSum()
393 const int64 output_elements = ShapeUtil::ElementsIn(convolution->shape()); in HandleConvolution() local395 output_elements * fmas_per_output_element * kFmaFlops; in HandleConvolution()