| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_ordering_test.cc | 140 auto cond_param = cond_builder.AddInstruction( local
141 HloInstruction::CreateParameter(0, scalar_shape, "cond_param"));
143 ShapeUtil::MakeShape(xla::PRED, {}), cond_param));
156 EXPECT_TRUE(ordering.ExecutesBefore(constant, cond_param));
164 EXPECT_FALSE(ordering.ExecutesBefore(xla_while, cond_param));
166 EXPECT_FALSE(ordering.ExecutesBefore(cond_param, xla_while));
169 EXPECT_TRUE(ordering.ExecutesBefore(cond_param, body_param));
171 EXPECT_TRUE(ordering.ExecutesBefore(cond_param, negate));
174 EXPECT_FALSE(ordering.ExecutesBefore(body_param, cond_param));
223 auto cond_param = cond_builder.AddInstruction local
[all...] |
| heap_simulator_test.cc | 49 HloInstruction* cond_param = cond_builder.AddInstruction( local
50 HloInstruction::CreateParameter(0, tuple_shape, "cond_param"));
52 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 0));
54 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 1));
55 // Free cond_param[] (16 bytes), Alloc PRED[] (1 byte)
91 {cond_param, cond_iter, cond_data, cond_lt});
111 // %WhileCond (cond_param: f32[4]) -> pred[] {
112 // %cond_param = f32[4]{0} parameter(0)
113 // %slice = f32[1]{0} slice(f32[4]{0} %cond_param), slice={[0:1]}
138 HloInstruction* cond_param = cond_builder.AddInstruction local
700 HloInstruction* cond_param = cond_builder.AddInstruction( local
[all...] |
| hlo_alias_analysis_test.cc | 354 auto cond_param = cond_builder.AddInstruction( local
394 GetValueDefinedAt(cond_param, {1}),
405 HloPosition{cond_param, {1}}, HloPosition{negate_2, {}}));
542 auto cond_param = cond_builder.AddInstruction( local
569 HloPosition{cond_param, {}}));
576 HloPosition{cond_param, {0}}));
583 HloPosition{body_tuple, {1}}, HloPosition{cond_param, {1}}));
593 GetValueDefinedAt(cond_param, {1}),
954 auto cond_param = cond_builder.AddInstruction( local
1053 auto cond_param = cond_builder.AddInstruction( local
[all...] |
| bfloat16_propagation_test.cc | 573 auto cond_param = builder_cond.AddInstruction( local
574 HloInstruction::CreateParameter(0, shape, "cond_param"));
576 builder_cond.AddInstruction(CreateDot(shape, cond_param, cond_param));
613 EXPECT_TRUE(OutputsBF16(cond_param));
633 auto cond_param = builder_cond.AddInstruction( local
634 HloInstruction::CreateParameter(0, shape, "cond_param"));
640 ShapeUtil::MakeShape(F32, {1, 1}), cond_param, {0, 0}, {1, 1},
645 ShapeUtil::MakeShape(F32, {1, 1}), cond_param, {1, 1}, {2, 2},
699 auto cond_param = builder_cond.AddInstruction local
[all...] |
| hlo_memory_scheduler_test.cc | 252 HloInstruction* cond_param = cond_builder.AddInstruction( local
253 HloInstruction::CreateParameter(0, r1f32, "cond_param"));
258 HloInstruction::CreateCompare(ShapeUtil::MakeShape(PRED, {}), cond_param,
|
| hlo_dataflow_analysis_test.cc | 501 auto cond_param = cond_builder.AddInstruction( local
530 EXPECT_FALSE(analysis.ValueIsDefinedAt(cond_param, /*index=*/{0}));
537 EXPECT_TRUE(analysis.ValueIsDefinedAt(cond_param, /*index=*/{1}));
538 EXPECT_TRUE(analysis.GetValueDefinedAt(cond_param, /*index=*/{1}).is_phi());
558 EXPECT_FALSE(analysis.ValueIsDefinedAt(cond_param, /*index=*/{0}));
559 EXPECT_FALSE(analysis.ValueIsDefinedAt(cond_param, /*index=*/{1}));
791 auto cond_param = cond_builder.AddInstruction( local
1348 auto cond_param = cond_builder.AddInstruction( local
[all...] |
