Home
last modified time | relevance | path

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

/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/
Dhlo_dataflow_analysis_test.cc1832 auto tuple_operand = builder.AddInstruction( in TEST_P() local
1835 scalar_shape_, pred, tuple_operand, true_computation, tuple_operand, in TEST_P()
1845 EXPECT_TRUE(analysis.ValueIsDefinedAt(tuple_operand)); in TEST_P()
1857 analysis.GetValueDefinedAt(tuple_operand)); in TEST_P()
1859 analysis.GetValueDefinedAt(tuple_operand)); in TEST_P()
1879 EXPECT_THAT(analysis.GetValueDefinedAt(tuple_operand).uses(), in TEST_P()
1967 auto tuple_operand = builder.AddInstruction( in TEST_P() local
1970 scalar_shape_, pred1, tuple_operand, inner_conditional_computation, in TEST_P()
1982 EXPECT_TRUE(analysis.ValueIsDefinedAt(tuple_operand)); in TEST_P()
2005 analysis.GetValueDefinedAt(tuple_operand)); in TEST_P()
Dconditional_code_motion.cc1076 const auto* tuple_operand = root->operand(gte->tuple_index()); in BenefitForMovingBoundaries() local
1077 if (tuple_operand->opcode() == HloOpcode::kCopy) { in BenefitForMovingBoundaries()
1078 if (Shape::Equal()(tuple_operand->operand(0)->shape(), in BenefitForMovingBoundaries()
Ddynamic_dimension_inference.cc1287 const HloInstruction* tuple_operand = hlo->operand(operand_index); in HandleConditional() local
1288 for (int64 i = 0; i < tuple_operand->operand_count(); ++i) { in HandleConditional()
1291 if (dynamic_size == tuple_operand->operand(i)) { in HandleConditional()