Searched refs:GetHloOutputs (Results 1 – 3 of 3) sorted by relevance
254 std::vector<mlir::Value> GetHloOutputs(mlir::Operation* op);
305 std::vector<mlir::Value> results = GetHloOutputs(reduce); in IsReductionFromOrToContiguousDimensions()692 std::vector<mlir::Value> GetHloOutputs(mlir::Operation* op) { in GetHloOutputs() function
302 } else if (opcode == HloOpcode::kReduce && GetHloOutputs(op).size() == 1) { in MayPreventVectorization()360 for (mlir::Value result : GetHloOutputs(op)) { in ComputeMaxUnrollFactor()456 for (auto result : GetHloOutputs(op)) { in GetIndexTypeForKernelFromMlir()698 outputs = GetHloOutputs(input.op); in EmitUsingElementalIrEmitter()2022 CHECK_EQ(1, GetHloOutputs(mlir_input.op).size()); in HandleFusion()2143 if (GetHloOutputs(mlir_input.op).size() == 1 && in EmitReduceFromMlir()3542 auto outputs = GetHloOutputs(op); in BuildKernelThunkForMlir()4231 auto results = GetHloOutputs(reduce_inst); in EmitPrologueForReduction()5855 auto outputs = GetHloOutputs(op); in SetOperation()