| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_copy_insertion_test.cc | 88 auto xla_while = builder.AddInstruction( local
97 EXPECT_THAT(xla_while->operand(0), op::Copy(op::Parameter()));
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_ordering.cc | 151 const HloInstruction* xla_while = use.instruction; local
153 xla_while->while_body()) ||
155 xla_while->while_condition())) {
166 const HloInstruction* xla_while = value.defining_instruction(); local
168 xla_while->while_body()) ||
170 xla_while->while_condition())) {
|
| hlo_ordering_test.cc | 196 auto xla_while = builder.AddInstruction( local
201 EXPECT_TRUE(ordering.ExecutesBefore(constant, xla_while));
209 EXPECT_FALSE(ordering.ExecutesBefore(xla_while, body_param));
210 EXPECT_FALSE(ordering.ExecutesBefore(xla_while, cond_param));
211 EXPECT_FALSE(ordering.ExecutesBefore(body_param, xla_while));
212 EXPECT_FALSE(ordering.ExecutesBefore(cond_param, xla_while));
259 auto xla_while = builder.AddInstruction( local
262 scalar_shape, HloOpcode::kAdd, constant, xla_while));
272 dataflow->GetValueDefinedAt(xla_while)));
275 dataflow->GetValueDefinedAt(xla_while), *dataflow))
[all...] |
| hlo_alias_analysis_test.cc | 360 auto xla_while = builder.AddInstruction( local
368 analysis.GetUniqueBufferAt(xla_while, /*index=*/{}).ComputePositions(),
369 UnorderedElementsAre(HloPosition{tuple, {}}, HloPosition{xla_while, {}},
374 analysis.GetUniqueBufferAt(xla_while, /*index=*/{0}).ComputePositions(),
377 HloPosition{xla_while, {0}}, HloPosition{body_param, {0}},
381 analysis.GetUniqueBufferAt(xla_while, /*index=*/{1}).ComputePositions(),
384 HloPosition{xla_while, {1}}, HloPosition{body_param, {1}},
389 GetValuesInBuffer(analysis.GetUniqueBufferAt(xla_while, /*index=*/{0})),
392 GetValuesInBuffer(analysis.GetUniqueBufferAt(xla_while, /*index=*/{1})),
394 GetValueDefinedAt(xla_while, /*index=*/{1})
624 auto xla_while = builder.AddInstruction( local
773 auto xla_while = builder.AddInstruction( local
864 auto xla_while = builder.AddInstruction( local
[all...] |
| copy_insertion_test.cc | 721 auto xla_while = BuildWhileInstructionWithCustomInit(loop_state_shape_, local
726 auto gte = xla_while->parent()->AddInstruction(
728 ShapeUtil::GetSubshape(xla_while->shape(), {1}), xla_while, 1));
729 auto sub = xla_while->parent()->AddInstruction(HloInstruction::CreateBinary(
731 auto gte0 = xla_while->parent()->AddInstruction(
733 ShapeUtil::GetSubshape(xla_while->shape(), {0}), xla_while, 0));
734 auto tuple = xla_while->parent()->AddInstruction(
737 xla_while->parent()->set_root_instruction(tuple)
1321 auto xla_while = builder.AddInstruction( local
1386 auto xla_while = builder.AddInstruction( local
1527 auto xla_while = builder.AddInstruction(HloInstruction::CreateWhile( local
1584 auto xla_while = builder.AddInstruction( local
1700 HloInstruction* xla_while = builder.AddInstruction( local
1765 HloInstruction* xla_while = builder.AddInstruction( local
[all...] |
| hlo_dataflow_analysis_test.cc | 507 auto xla_while = builder.AddInstruction( local
519 EXPECT_FALSE(analysis.ValueIsDefinedAt(xla_while, /*index=*/{0}));
524 EXPECT_TRUE(analysis.ValueIsDefinedAt(xla_while, /*index=*/{1}));
525 EXPECT_TRUE(analysis.GetValueDefinedAt(xla_while, /*index=*/{1}).is_phi());
534 HloUse{xla_while, 0, {0}}));
538 EXPECT_TRUE(analysis.GetValueDefinedAt(xla_while, /*index=*/{1})
545 EXPECT_FALSE(analysis.ValueIsDefinedAt(xla_while, /*index=*/{0}));
546 EXPECT_FALSE(analysis.ValueIsDefinedAt(xla_while, /*index=*/{1}));
794 auto xla_while = builder.AddInstruction( local
1051 auto xla_while = builder.AddInstruction( local
1330 auto xla_while = builder.AddInstruction( local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| while_test.cc | 451 auto xla_while = builder.While(condition, body, init); local
453 auto add12 = builder.Add(builder.GetTupleElement(xla_while, 1),
454 builder.GetTupleElement(xla_while, 2));
455 auto result = builder.Add(add12, builder.GetTupleElement(xla_while, 3));
[all...] |
