/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...] |