Searched refs:first_reduce (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | ir_emission_utils.h | 231 const HloInstruction* first_reduce); 233 mlir::mhlo::ReduceOp first_reduce); 237 const HloInstruction* first_reduce) { in AreFusedReductionOutputsConsistent() argument 239 return IsFusedReductionOutputConsistent(inst, first_reduce); in AreFusedReductionOutputsConsistent()
|
D | ir_emission_utils.cc | 595 const HloInstruction* first_reduce) { in IsFusedReductionOutputConsistent() argument 600 return ShapeUtil::Equal(first_reduce->shape(), inst->shape()) && in IsFusedReductionOutputConsistent() 601 ShapeUtil::Equal(first_reduce->operand(0)->shape(), in IsFusedReductionOutputConsistent() 603 ShapeUtil::Equal(first_reduce->operand(1)->shape(), in IsFusedReductionOutputConsistent() 605 first_reduce->dimensions() == inst->dimensions(); in IsFusedReductionOutputConsistent() 608 first_reduce->operand(0)->shape(), inst->shape()) && in IsFusedReductionOutputConsistent() 609 LayoutUtil::Equal(first_reduce->operand(0)->shape().layout(), in IsFusedReductionOutputConsistent() 614 mlir::mhlo::ReduceOp first_reduce) { in IsFusedReductionOutputConsistent() argument 615 CHECK_EQ(1, first_reduce.getNumResults()); in IsFusedReductionOutputConsistent() 617 TypeToShape(first_reduce.operands()[0].getType()); in IsFusedReductionOutputConsistent() [all …]
|
D | ir_emitter_unnested.cc | 4222 mlir::Operation* first_reduce = nullptr; in EmitPrologueForReduction() local 4236 if (first_reduce == nullptr) { in EmitPrologueForReduction() 4237 first_reduce = reduce_inst; in EmitPrologueForReduction() 4240 first_reduce->getAttrOfType<mlir::DenseIntElementsAttr>("dimensions"), in EmitPrologueForReduction() 4315 CHECK(first_reduce); in EmitPrologueForReduction() 5295 mlir::Operation* unnested_hlo, mlir::Operation* first_reduce) { in ComputeReductionCodegenInfo() argument 5296 Shape input_shape = TypeToShape(first_reduce->getOperand(0).getType()); in ComputeReductionCodegenInfo() 5298 GetReductionKindAndContiguousComponents(first_reduce); in ComputeReductionCodegenInfo() 5310 int smallest_input_dtype_bits = get_dtype_bits(first_reduce->getOperand(0)); in ComputeReductionCodegenInfo() 5569 mlir::Operation* first_reduce = nullptr; in EmitReductionFromOrToContiguousDimensions() local [all …]
|
D | ir_emitter_unnested.h | 445 mlir::Operation* unnested_hlo, mlir::Operation* first_reduce);
|