/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | gpu_fusible_test.cc | 48 module->entry_computation()->root_instruction(); in TEST_F() 51 module->entry_computation()->root_instruction()->operand(0); in TEST_F() 84 module->entry_computation()->root_instruction()->operand(0); in TEST_F() 88 module->entry_computation()->root_instruction()->operand(1); in TEST_F() 122 module->entry_computation()->root_instruction()->operand(0); in TEST_F() 126 module->entry_computation()->root_instruction()->operand(1); in TEST_F() 147 module->entry_computation()->root_instruction(); in TEST_F() 150 module->entry_computation()->root_instruction()->operand(0); in TEST_F() 182 module->entry_computation()->root_instruction(); in TEST_F() 186 module->entry_computation()->root_instruction()->operand(0); in TEST_F() [all …]
|
D | instruction_fusion_test.cc | 47 EXPECT_EQ(broadcast2, computation->root_instruction()); in TEST_F() 51 EXPECT_EQ(broadcast2, computation->root_instruction()); in TEST_F() 67 EXPECT_EQ(broadcast2, computation->root_instruction()); in TEST_F() 71 EXPECT_THAT(computation->root_instruction(), op::Fusion()); in TEST_F() 86 EXPECT_EQ(reshape2, computation->root_instruction()); in TEST_F() 90 EXPECT_THAT(computation->root_instruction(), op::Fusion()); in TEST_F() 105 EXPECT_EQ(transpose2, computation->root_instruction()); in TEST_F() 109 EXPECT_THAT(computation->root_instruction(), op::Fusion()); in TEST_F() 125 EXPECT_EQ(log, computation->root_instruction()); in TEST_F() 142 EXPECT_EQ(transpose2, computation->root_instruction()); in TEST_F() [all …]
|
D | cudnn_pad_for_convolutions_test.cc | 47 auto* root = module->entry_computation()->root_instruction(); in TEST_F() 72 auto* root = module->entry_computation()->root_instruction(); in TEST_F() 95 auto* root = module->entry_computation()->root_instruction(); in TEST_F() 116 auto* root = module->entry_computation()->root_instruction(); in TEST_F() 138 auto* root = module->entry_computation()->root_instruction(); in TEST_F() 160 auto* root = module->entry_computation()->root_instruction(); in TEST_F() 181 auto* root = module->entry_computation()->root_instruction(); in TEST_F() 206 auto* root = module->entry_computation()->root_instruction(); in TEST_F() 232 auto* root = module->entry_computation()->root_instruction(); in TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | ar_crs_combiner_test.cc | 43 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F() 63 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F() 82 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F() 103 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F() 123 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F() 143 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F() 165 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F() 187 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F() 210 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F() 246 auto root_while = module->entry_computation()->root_instruction(); in TEST_F() [all …]
|
D | reshape_mover_test.cc | 56 EXPECT_THAT(computation->root_instruction(), in TEST_F() 61 EXPECT_THAT(computation->root_instruction(), in TEST_F() 99 EXPECT_THAT(computation->root_instruction(), 104 EXPECT_THAT(computation->root_instruction(), 125 EXPECT_THAT(computation->root_instruction(), in TEST_F() 131 computation->root_instruction(), in TEST_F() 152 EXPECT_THAT(computation->root_instruction(), in TEST_F() 156 EXPECT_THAT(computation->root_instruction(), in TEST_F() 159 computation->root_instruction()->shape().DebugString()); in TEST_F() 206 EXPECT_THAT(computation->root_instruction(), [all …]
|
D | root_instruction_sinker_test.cc | 58 module->entry_computation()->root_instruction()->while_body(); in TEST_F() 63 ->root_instruction() in TEST_F() 104 module->entry_computation()->root_instruction()->while_body(); in TEST_F() 107 while_body->root_instruction()); in TEST_F() 108 EXPECT_THAT(while_body->root_instruction(), in TEST_F() 129 module->entry_computation()->root_instruction()->called_computations()[0]; in TEST_F() 134 ->root_instruction() in TEST_F() 161 module->entry_computation()->root_instruction()->called_computations()[0]; in TEST_F() 164 called_computation->root_instruction()); in TEST_F() 165 EXPECT_THAT(called_computation->root_instruction(), in TEST_F()
|
D | algebraic_simplifier_test.cc | 70 HloInstruction* root = computation->root_instruction(); in TEST_F() 74 root = computation->root_instruction(); in TEST_F() 94 m->entry_computation()->root_instruction(), in TEST_F() 114 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F() 132 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F() 153 m->entry_computation()->root_instruction(), in TEST_F() 173 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F() 196 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F() 218 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F() 276 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F() [all …]
|
D | stable_sort_expander_test.cc | 107 auto root = module->entry_computation()->root_instruction(); in TEST_F() 111 root->operand(0)->to_apply()->root_instruction(), /*iota_parameter=*/1); in TEST_F() 153 auto root = module->entry_computation()->root_instruction(); in TEST_F() 157 root->operand(0)->to_apply()->root_instruction(), /*iota_parameter=*/1); in TEST_F() 183 auto root = module->entry_computation()->root_instruction(); in TEST_F() 191 root->operand(0)->operand(0)->to_apply()->root_instruction(), in TEST_F() 251 auto root = module->entry_computation()->root_instruction(); in TEST_F() 255 root->operand(0)->to_apply()->root_instruction(), in TEST_F() 288 auto root = module->entry_computation()->root_instruction(); in TEST_F() 292 root->operand(0)->to_apply()->root_instruction(), in TEST_F() [all …]
|
D | hlo_constant_folding_test.cc | 55 EXPECT_THAT(computation->root_instruction(), in TEST_F() 62 EXPECT_THAT(computation->root_instruction(), GmockMatch(m::Constant())); in TEST_F() 63 EXPECT_EQ(computation->root_instruction()->literal().GetFirstElement<int64>(), in TEST_F() 77 EXPECT_THAT(computation->root_instruction(), in TEST_F() 84 EXPECT_THAT(computation->root_instruction(), GmockMatch(m::Constant())); in TEST_F() 85 EXPECT_EQ(computation->root_instruction()->literal().GetFirstElement<float>(), in TEST_F() 99 EXPECT_THAT(computation->root_instruction(), in TEST_F() 106 EXPECT_THAT(computation->root_instruction(), GmockMatch(m::Constant())); in TEST_F() 107 EXPECT_EQ(computation->root_instruction()->literal().Get<int64>({0}), 42); in TEST_F() 108 EXPECT_EQ(computation->root_instruction()->literal().Get<int64>({1}), 19); in TEST_F() [all …]
|
D | hlo_cost_analysis_test.cc | 148 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F() 159 HloInstruction* root = hlo_module->entry_computation()->root_instruction(); in TEST_F() 182 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F() 193 HloInstruction* root = hlo_module->entry_computation()->root_instruction(); in TEST_F() 218 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F() 229 HloInstruction* root = hlo_module->entry_computation()->root_instruction(); in TEST_F() 248 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F() 259 HloInstruction* root = hlo_module->entry_computation()->root_instruction(); in TEST_F() 275 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F() 282 HloInstruction* root = hlo_module->entry_computation()->root_instruction(); in TEST_F() [all …]
|
D | while_loop_simplifier_test.cc | 132 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F() 140 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F() 147 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F() 155 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F() 168 auto* while_op = computation->root_instruction(); in TEST_F() 173 while_op->while_body()->root_instruction())); in TEST_F() 175 EXPECT_THAT(computation->root_instruction()->control_predecessors(), in TEST_F() 185 auto* while_op = computation->root_instruction(); in TEST_F() 201 auto* while_op = computation->root_instruction(); in TEST_F() 217 auto* while_op = computation->root_instruction(); in TEST_F() [all …]
|
D | convolution_4d_expander_test.cc | 45 HloInstruction* root = computation->root_instruction(); in TEST_F() 50 root = computation->root_instruction(); in TEST_F() 70 HloInstruction* root = computation->root_instruction(); in TEST_F() 75 root = computation->root_instruction(); in TEST_F() 97 HloInstruction* root = computation->root_instruction(); in TEST_F() 102 root = computation->root_instruction(); in TEST_F() 122 HloInstruction* root = computation->root_instruction(); in TEST_F() 141 HloInstruction* root = computation->root_instruction(); in TEST_F() 160 HloInstruction* root = computation->root_instruction(); in TEST_F()
|
D | call_inliner_test.cc | 69 EXPECT_THAT(computation->root_instruction(), op::Constant()); in TEST_F() 70 EXPECT_EQ(computation->root_instruction()->literal().GetFirstElement<float>(), in TEST_F() 72 ASSERT_EQ(1, computation->root_instruction()->control_predecessors().size()); in TEST_F() 73 auto prior = computation->root_instruction()->control_predecessors()[0]; in TEST_F() 113 computation->root_instruction()->while_condition()->root_instruction(), in TEST_F() 115 EXPECT_THAT(computation->root_instruction()->while_body()->root_instruction(), in TEST_F() 140 EXPECT_THAT(computation->root_instruction(), op::Constant()); in TEST_F() 141 EXPECT_THAT(computation->root_instruction()->control_successors(), in TEST_F() 182 EXPECT_THAT(computation->root_instruction(), op::Constant()); in TEST_F() 200 outfeed_computation->root_instruction()->shape(), /*operands=*/{}, in TEST_F()
|
D | logistic_expander_test.cc | 63 HloInstruction* root = computation->root_instruction(); in TEST_F() 67 root = computation->root_instruction(); in TEST_F() 68 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F() 90 HloInstruction* root = computation->root_instruction(); in TEST_F() 94 root = computation->root_instruction(); in TEST_F() 95 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F()
|
D | while_loop_analysis_test.cc | 55 HloInstruction* while_op = module->entry_computation()->root_instruction(); in TEST_F() 86 HloInstruction* while_op = module->entry_computation()->root_instruction(); in TEST_F() 119 HloInstruction* while_op = module->entry_computation()->root_instruction(); in TEST_F() 154 HloInstruction* while_op = module->entry_computation()->root_instruction(); in TEST_F() 193 HloInstruction* while_op = module->entry_computation()->root_instruction(); in TEST_F() 232 HloInstruction* while_op = module->entry_computation()->root_instruction(); in TEST_F() 267 HloInstruction* while_op = module->entry_computation()->root_instruction(); in TEST_F()
|
D | while_loop_constant_sinking_test.cc | 64 EXPECT_THAT(while_body->root_instruction(), in TEST_F() 104 EXPECT_THAT(while_body->root_instruction(), in TEST_F() 145 EXPECT_THAT(while_body->root_instruction(), in TEST_F() 193 EXPECT_THAT(while_body->root_instruction(), in TEST_F() 235 EXPECT_THAT(while_body->root_instruction(), in TEST_F() 281 EXPECT_THAT(while_condition->root_instruction(), op::Lt(_, op::Constant())); in TEST_F() 321 EXPECT_THAT(while_condition->root_instruction(), in TEST_F() 364 EXPECT_THAT(while_condition->root_instruction(), op::Lt(_, op::Constant())); in TEST_F() 414 EXPECT_THAT(while_condition->root_instruction(), in TEST_F()
|
D | conditional_to_select_test.cc | 72 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() 75 EXPECT_THAT(mapped->root_instruction(), in TEST_F() 125 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() 129 mapped->root_instruction(), in TEST_F() 181 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() 184 EXPECT_THAT(mapped->root_instruction(), in TEST_F()
|
D | convolution_group_converter_test.cc | 49 HloInstruction* root = computation->root_instruction(); in TEST_F() 55 root = computation->root_instruction(); in TEST_F() 81 HloInstruction* root = computation->root_instruction(); in TEST_F() 87 root = computation->root_instruction(); in TEST_F() 108 HloInstruction* root = computation->root_instruction(); in TEST_F() 114 root = computation->root_instruction(); in TEST_F() 135 HloInstruction* root = computation->root_instruction(); in TEST_F()
|
D | tuple_simplifier_test.cc | 104 EXPECT_THAT(computation->root_instruction(), gte); in TEST_F() 108 EXPECT_THAT(computation->root_instruction(), param1); in TEST_F() 131 EXPECT_THAT(computation->root_instruction(), in TEST_F() 136 EXPECT_THAT(computation->root_instruction(), op::Negate(op::Parameter())); in TEST_F() 163 EXPECT_THAT(computation->root_instruction(), element); in TEST_F() 167 EXPECT_THAT(computation->root_instruction(), param); in TEST_F() 188 EXPECT_THAT(computation->root_instruction(), tuple); in TEST_F() 192 EXPECT_THAT(computation->root_instruction(), tuple_param); in TEST_F() 213 EXPECT_THAT(computation->root_instruction(), tuple); in TEST_F() 217 EXPECT_THAT(computation->root_instruction(), tuple); in TEST_F() [all …]
|
D | while_util_test.cc | 77 HloInstruction* while_instr = entry_computation->root_instruction(); in TEST_F() 87 entry_computation->root_instruction(), in TEST_F() 95 EXPECT_THAT(new_while_instr->while_body()->root_instruction(), in TEST_F() 108 HloInstruction* while_instr = entry_computation->root_instruction(); in TEST_F() 121 entry_computation->root_instruction(), in TEST_F() 129 EXPECT_THAT(new_while_instr->while_body()->root_instruction(), in TEST_F() 204 HloInstruction* while_instr = main->root_instruction(); in TEST_F()
|
D | topk_rewriter_test.cc | 85 module->entry_computation()->root_instruction(), in TEST_F() 89 module->entry_computation()->root_instruction()->operand(0)->operand(0); in TEST_F() 115 module->entry_computation()->root_instruction(), in TEST_F() 119 module->entry_computation()->root_instruction()->operand(0)->operand(0); in TEST_F() 146 module->entry_computation()->root_instruction(), in TEST_F() 152 ->root_instruction() in TEST_F()
|
D | defuser_test.cc | 75 EXPECT_THAT(computation->root_instruction(), op::Fusion()); in TEST_F() 81 EXPECT_THAT(computation->root_instruction(), in TEST_F() 101 EXPECT_THAT(computation->root_instruction(), op::Negate(op::Fusion())); in TEST_F() 107 EXPECT_THAT(computation->root_instruction(), in TEST_F() 140 EXPECT_THAT(computation->root_instruction(), op::Fusion()); in TEST_F() 146 EXPECT_THAT(computation->root_instruction(), in TEST_F() 180 EXPECT_THAT(computation->root_instruction(), op::Fusion()); in TEST_F() 186 EXPECT_THAT(computation->root_instruction(), in TEST_F() 210 EXPECT_THAT(computation->root_instruction(), op::Fusion()); in TEST_F() 216 EXPECT_THAT(computation->root_instruction(), op::Negate(op::Add())); in TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/tools/ |
D | hlo_extractor_test.cc | 45 EXPECT_THAT(extracted_module->entry_computation()->root_instruction(), in TEST_F() 52 EXPECT_THAT(extracted_module->entry_computation()->root_instruction(), in TEST_F() 59 EXPECT_THAT(extracted_module->entry_computation()->root_instruction(), in TEST_F() 83 EXPECT_THAT(extracted_module->entry_computation()->root_instruction(), in TEST_F() 90 EXPECT_THAT(extracted_module->entry_computation()->root_instruction(), in TEST_F() 96 EXPECT_THAT(extracted_module->entry_computation()->root_instruction(), in TEST_F() 103 EXPECT_THAT(extracted_module->entry_computation()->root_instruction(), in TEST_F() 127 EXPECT_THAT(extracted_module->entry_computation()->root_instruction(), in TEST_F() 133 EXPECT_THAT(extracted_module->entry_computation()->root_instruction(), in TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
D | spmd_partitioner_test.cc | 99 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() 117 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() 138 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() 162 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() 184 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() 207 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() 227 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() 251 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() 281 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() 308 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() [all …]
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | xla_builder_test.cc | 76 auto root = module->entry_computation()->root_instruction(); in TEST_F() 87 auto root = module->entry_computation()->root_instruction(); in TEST_F() 101 auto root = module->entry_computation()->root_instruction(); in TEST_F() 132 auto root = module->entry_computation()->root_instruction(); in TEST_F() 149 module->entry_computation()->root_instruction(), in TEST_F() 165 module->entry_computation()->root_instruction(), in TEST_F() 187 auto root = module->entry_computation()->root_instruction(); in TEST_F() 203 auto root = module->entry_computation()->root_instruction(); in TEST_F() 212 auto root = module->entry_computation()->root_instruction(); in TEST_F() 263 auto root = module->entry_computation()->root_instruction(); in TEST_F() [all …]
|