Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dfused_ir_emitter.cc107 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/
Dconditional_test.cc653 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/
Ddynamic_dimension_inference.cc471 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()
Dhlo_dataflow_analysis_test.cc1705 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()