Searched refs:tuple_operand (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
D | fused_ir_emitter.cc | 107 const HloInstruction* tuple_operand = get_tuple_element->operand(0); in HandleGetTupleElement() local 109 if (tuple_operand->opcode() == HloOpcode::kGetTupleElement) { in HandleGetTupleElement() 110 TF_ASSIGN_OR_RETURN(tuple_ptr, non_indexed_generators_[tuple_operand]()); in HandleGetTupleElement() 112 if (tuple_operand->opcode() != HloOpcode::kParameter) { in HandleGetTupleElement() 117 tuple_operand->name()); in HandleGetTupleElement() 120 GetBasePointerForFusedParameter(tuple_operand->parameter_number()); in HandleGetTupleElement()
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | conditional_test.cc | 653 auto tuple_operand = Tuple(&builder, {pred2, operand1, operand2}); in XLA_TEST_F() local 654 Conditional(pred1, tuple_operand, inner_builder_result.ConsumeValueOrDie(), in XLA_TEST_F() 681 auto tuple_operand = Tuple(&builder, {pred, operand1, operand2}); in XLA_TEST_F() local 682 Call(&builder, inner_builder_result.ConsumeValueOrDie(), {tuple_operand}); in XLA_TEST_F() 742 auto tuple_operand = Tuple(&builder, {x, y}); in XLA_TEST_F() local 743 Call(&builder, main, {tuple_operand}); in XLA_TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | dynamic_dimension_inference.cc | 471 const HloInstruction* tuple_operand = hlo->operand(0); in HandleWhile() local 472 for (int64 i = 0; i < tuple_operand->operand_count(); ++i) { in HandleWhile() 473 if (dynamic_size == tuple_operand->operand(i)) { in HandleWhile()
|
D | hlo_dataflow_analysis_test.cc | 1705 auto tuple_operand = builder.AddInstruction( in TEST_P() local 1708 scalar_shape_, pred, tuple_operand, true_computation, tuple_operand, in TEST_P() 1718 EXPECT_TRUE(analysis.ValueIsDefinedAt(tuple_operand)); in TEST_P() 1730 analysis.GetValueDefinedAt(tuple_operand)); in TEST_P() 1732 analysis.GetValueDefinedAt(tuple_operand)); in TEST_P() 1752 EXPECT_THAT(analysis.GetValueDefinedAt(tuple_operand).uses(), in TEST_P() 1840 auto tuple_operand = builder.AddInstruction( in TEST_P() local 1843 scalar_shape_, pred1, tuple_operand, inner_conditional_computation, in TEST_P() 1855 EXPECT_TRUE(analysis.ValueIsDefinedAt(tuple_operand)); in TEST_P() 1878 analysis.GetValueDefinedAt(tuple_operand)); in TEST_P()
|