Home
last modified time | relevance | path

Searched refs:GetHloOutputs (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emission_utils.h254 std::vector<mlir::Value> GetHloOutputs(mlir::Operation* op);
Dir_emission_utils.cc305 std::vector<mlir::Value> results = GetHloOutputs(reduce); in IsReductionFromOrToContiguousDimensions()
692 std::vector<mlir::Value> GetHloOutputs(mlir::Operation* op) { in GetHloOutputs() function
Dir_emitter_unnested.cc302 } 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()