Home
last modified time | relevance | path

Searched defs:get_tuple_element (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dfused_ir_emitter.cc112 const HloInstruction* get_tuple_element) { in HandleGetTupleElement()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_control_flow.cc201 auto get_tuple_element = builder->create<mhlo::GetTupleElementOp>( in ReplaceImplicitInputs() local
Dlegalize_tf_communication.cc294 auto get_tuple_element = in CreateRecvOp() local
/external/tensorflow/tensorflow/compiler/xla/service/
Ddfs_hlo_visitor_with_default.h152 Status HandleGetTupleElement(HloInstructionPtr get_tuple_element) override { in HandleGetTupleElement()
Dtuple_points_to_analysis.cc223 HloInstruction* get_tuple_element) { in HandleGetTupleElement()
Dtuple_points_to_analysis_test.cc253 auto get_tuple_element = builder.AddInstruction( in TEST_F() local
Dhlo_cost_analysis.cc229 const HloInstruction* get_tuple_element) { in HandleGetTupleElement()
Dhlo_verifier.cc634 Status ShapeVerifier::HandleGetTupleElement(HloInstruction* get_tuple_element) { in HandleGetTupleElement()
Dhlo_evaluator.cc1851 Status HloEvaluator::HandleGetTupleElement(HloInstruction* get_tuple_element) { in HandleGetTupleElement()
Dalgebraic_simplifier.cc2979 HloInstruction* get_tuple_element) { in HandleGetTupleElement()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.cc182 Status IrEmitter::HandleGetTupleElement(HloInstruction* get_tuple_element) { in HandleGetTupleElement()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc356 Status IrEmitter::HandleGetTupleElement(HloInstruction* get_tuple_element) { in HandleGetTupleElement()