Searched refs:MakeGetTupleElementHlo (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_util.cc | 172 MakeGetTupleElementHlo(param, 0)); in MakeCountedLoopConditionComputation()
192 MakeGetTupleElementHlo(param, 0)); in MakeCountedLoopBodyComputation()
199 MakeGetTupleElementHlo(param, i)); in MakeCountedLoopBodyComputation()
|
| D | reduce_decomposer.cc | 128 return MakeGetTupleElementHlo(instr, idx); in GetOutput()
|
| D | scatter_simplifier.cc | 196 MakeGetTupleElementHlo(result, i)); in ExpandInstruction()
|
| D | hlo_creation_utils.h | 128 StatusOr<HloInstruction*> MakeGetTupleElementHlo(
|
| D | hlo_creation_utils.cc | 259 StatusOr<HloInstruction*> MakeGetTupleElementHlo(HloInstruction* operand, in MakeGetTupleElementHlo() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | triangular_solve_rewriter.cc | 66 MakeGetTupleElementHlo(custom_call, 0)); in Run()
|
| D | cudnn_fused_conv_rewriter.cc | 177 MakeGetTupleElementHlo(new_conv, 0)); in FuseConvertToFloat()
326 MakeGetTupleElementHlo(new_conv, 0)); in FuseBiasOrSideInput()
536 MakeGetTupleElementHlo(new_conv, 0)); in FuseConvertToF16()
617 MakeGetTupleElementHlo(new_conv, 0)); in FuseConvertToS8()
|
| D | horizontal_loop_fusion.cc | 474 MakeGetTupleElementHlo(hori_fusion_instr, total_output_id++)); in Fuse()
|