Searched refs:first_reduce (Results 1 – 2 of 2) sorted by relevance
2711 const HloInstruction* first_reduce = nullptr; in EmitPrologueForReduction() local2717 if (first_reduce == nullptr) { in EmitPrologueForReduction()2718 first_reduce = reduce_inst; in EmitPrologueForReduction()2720 CHECK(first_reduce->dimensions() == reduce_inst->dimensions()); in EmitPrologueForReduction()3466 const HloInstruction* first_reduce) { in AreFusedReductionOutputsConsistent() argument3471 TF_RET_CHECK(ShapeUtil::Equal(first_reduce->shape(), inst->shape())); in AreFusedReductionOutputsConsistent()3472 TF_RET_CHECK(ShapeUtil::Equal(first_reduce->operand(0)->shape(), in AreFusedReductionOutputsConsistent()3474 TF_RET_CHECK(ShapeUtil::Equal(first_reduce->operand(1)->shape(), in AreFusedReductionOutputsConsistent()3476 TF_RET_CHECK(first_reduce->dimensions() == inst->dimensions()); in AreFusedReductionOutputsConsistent()3482 first_reduce->operand(0)->shape(), inst->shape())); in AreFusedReductionOutputsConsistent()[all …]
227 const HloInstruction* first_reduce);