Home
last modified time | relevance | path

Searched refs:first_reduce (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emission_utils.h231 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()
Dir_emission_utils.cc595 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 …]
Dir_emitter_unnested.cc4222 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 …]
Dir_emitter_unnested.h445 mlir::Operation* unnested_hlo, mlir::Operation* first_reduce);