Home
last modified time | relevance | path

Searched defs:custom_call_instr (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_instruction_test.cc2065 auto custom_call_instr = Cast<HloCustomCallInstruction>(instr.get()); in TEST_F() local
2078 auto custom_call_instr = Cast<HloCustomCallInstruction>(instr.get()); in TEST_F() local
H A Dhlo_parser.cc2700 auto custom_call_instr = Cast<HloCustomCallInstruction>(instruction); in CreateInstruction() local
H A Dhlo_instruction.cc749 auto custom_call_instr = in CreateFromProto() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dmhlo_to_lhlo_with_xla.cc696 auto* custom_call_instr = xla::Cast<xla::HloCustomCallInstruction>(instr); in EmitCustomCallOp() local