Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emission_utils.h253 std::vector<mlir::Value> GetHloOperands(mlir::Operation* op);
Dir_emission_utils.cc672 std::vector<mlir::Value> GetHloOperands(mlir::Operation* op) { in GetHloOperands() function
Dir_emitter_unnested.cc697 auto operands = GetHloOperands(input.op); in EmitUsingElementalIrEmitter()
3541 auto operands = GetHloOperands(op); in BuildKernelThunkForMlir()
5312 for (mlir::Value operand : GetHloOperands(unnested_hlo)) { in ComputeReductionCodegenInfo()
5854 auto operands = GetHloOperands(op); in SetOperation()