Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_utils.h90 ::xla::StatusOr<::xla::HloOpcode> MhloToHloOpcode(mlir::Operation* op);
Dhlo_utils.cc284 StatusOr<::xla::HloOpcode> MhloToHloOpcode(mlir::Operation* op) { in MhloToHloOpcode() function
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnccl_all_reduce_thunk.cc59 StatusOr<HloOpcode> opcode = MhloToHloOpcode(reduction_op); in MatchReductionComputation()
Dir_emitter_unnested.cc236 auto opcode = *MhloToHloOpcode(op); in LmhloOpIsElementwise()
259 auto opcode = *MhloToHloOpcode(op); in MayPreventVectorization()
269 switch (*MhloToHloOpcode(&instr)) { in MayPreventVectorization()
1826 HloOpcode opcode = *MhloToHloOpcode(&op); in EmitLoopFusionFromMlir()
5021 opcode = *MhloToHloOpcode(op); in IsInstructionSafeForShmemTranspose()
5197 (HloInstruction::IsOpElementwise(*MhloToHloOpcode(user)) && in AreUsersElementwise()