| /external/tensorflow/tensorflow/compiler/xla/service/ |
| 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...] |
| 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...] |
| 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())) {
|
| copy_insertion.cc | 140 const HloInstruction* xla_while,
142 DCHECK(ShapeUtil::Compatible(indices_to_copy->shape(), xla_while->shape()));
145 const HloInstruction* init = xla_while->operand(0);
151 dataflow.GetValueSet(xla_while, index).values().size() > 1) {
157 should_copy = dataflow.GetUniqueValueAt(xla_while, index) !=
228 HloInstruction* xla_while) {
229 VLOG(2) << "Adding copies for kWhile instruction " << xla_while->name();
230 TF_RET_CHECK(xla_while->opcode() == HloOpcode::kWhile);
232 ShapeTree<bool> indices_to_copy(xla_while->shape());
233 if (!IndicesToCopyForWhile(alias_analysis.dataflow_analysis(), xla_while,
[all...] |
| dfs_hlo_visitor_with_default.h | 173 Status HandleWhile(HloInstructionPtr xla_while) override {
174 return DefaultAction(xla_while);
|
| 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.cc | 539 bool HloDataflowAnalysis::UpdateWhileValueSet(HloInstruction* xla_while) {
540 CHECK_EQ(xla_while->opcode(), HloOpcode::kWhile);
542 &GetInstructionValueSet(xla_while->while_body()->root_instruction()),
543 &GetInstructionValueSet(xla_while->operand(0))};
545 return Phi(xla_while, inputs);
547 return GetInstructionValueSet(xla_while).AssignUnionOf(inputs);
[all...] |
| hlo_verifier.h | 72 Status HandleWhile(HloInstruction* xla_while) override;
|
| hlo_cost_analysis.h | 101 Status HandleWhile(const HloInstruction* xla_while) override;
|
| hlo_dataflow_analysis.h | 158 bool UpdateWhileValueSet(HloInstruction* xla_while);
|
| hlo_cost_analysis.cc | 493 Status HloCostAnalysis::HandleWhile(const HloInstruction* xla_while) {
500 ProcessSubcomputation(xla_while->while_body()));
503 ProcessSubcomputation(xla_while->while_condition()));
|
| hlo_verifier.cc | 282 Status ShapeVerifier::HandleWhile(HloInstruction* xla_while) {
285 return CheckShape(xla_while,
286 xla_while->while_body()->ComputeProgramShape().result());
[all...] |
| /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()));
|
| ir_emitter.cc | [all...] |
| cpu_compiler.cc | 214 Status HandleWhile(HloInstruction* xla_while) override {
215 TF_RETURN_IF_ERROR(DefaultAction(xla_while));
220 xla_while->while_condition()->Accept(&candidates_for_condition));
224 TF_RETURN_IF_ERROR(xla_while->while_body()->Accept(&candidates_for_body));
[all...] |
| ir_emitter.h | 150 Status HandleWhile(HloInstruction* xla_while) override;
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| ir_emitter_unnested.cc | [all...] |
| ir_emitter_unnested.h | 75 Status HandleWhile(HloInstruction* xla_while) override;
|
| /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...] |
