Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_util.cc167 MakeGetTupleElementHlo(param, 0)); in MakeCountedLoopConditionComputation()
187 MakeGetTupleElementHlo(param, 0)); in MakeCountedLoopBodyComputation()
194 MakeGetTupleElementHlo(param, i)); in MakeCountedLoopBodyComputation()
268 MakeGetTupleElementHlo(while_instr, i + 1)); in MakeCountedLoop()
Dhlo_creation_utils.h112 StatusOr<HloInstruction*> MakeGetTupleElementHlo(HloInstruction* operand,
Dhlo_creation_utils.cc222 StatusOr<HloInstruction*> MakeGetTupleElementHlo(HloInstruction* operand, in MakeGetTupleElementHlo() function
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhorizontal_loop_fusion.cc452 TF_ASSIGN_OR_RETURN(auto gep, MakeGetTupleElementHlo(hori_fusion_instr, in Fuse()