/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | hlo_creation_utils_test.cc | 35 HloComputation** entry_computation) { in CreateModuleWithProgramShape() argument 40 *entry_computation = module->AddEntryComputation( in CreateModuleWithProgramShape() 43 *param = (*entry_computation)->parameter_instruction(0); in CreateModuleWithProgramShape() 50 HloComputation** entry_computation, PrimitiveType primitive_type_output) { in CreateModuleWithProgramShape() argument 55 *entry_computation = module->AddEntryComputation( in CreateModuleWithProgramShape() 58 *param = (*entry_computation)->parameter_instruction(0); in CreateModuleWithProgramShape() 65 HloComputation* entry_computation; in TEST_F() local 69 &entry_computation); in TEST_F() 73 entry_computation->set_root_instruction(first_1_dims_collapsed); in TEST_F() 84 HloComputation* entry_computation; in TEST_F() local [all …]
|
H A D | ar_crs_combiner_test.cc | 44 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F() 48 i1, module->entry_computation()->parameter_instruction(0))); in TEST_F() 64 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F() 83 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F() 104 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F() 124 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F() 144 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F() 166 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F() 188 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F() 211 auto root_tuple = module->entry_computation()->root_instruction(); in TEST_F() [all …]
|
H A D | hlo_module_dce_test.cc | 112 EXPECT_FALSE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F() 114 EXPECT_FALSE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F() 152 EXPECT_FALSE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F() 154 EXPECT_FALSE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F() 190 EXPECT_FALSE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F() 193 EXPECT_FALSE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F() 196 auto while_loops = GetWhileLoops(module->entry_computation()); in TEST_F() 234 EXPECT_FALSE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F() 237 EXPECT_FALSE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F() 240 EXPECT_FALSE(WhileBodyHasPassThroughTupleElement(module->entry_computation(), in TEST_F() [all …]
|
H A D | while_loop_all_reduce_code_motion_test.cc | 95 *(std::find_if(module->entry_computation()->instructions().begin(), in TEST_F() 96 module->entry_computation()->instructions().end(), in TEST_F() 108 *(std::find_if(module->entry_computation()->instructions().begin(), in TEST_F() 109 module->entry_computation()->instructions().end(), in TEST_F() 194 *(std::find_if(module->entry_computation()->instructions().begin(), in TEST_F() 195 module->entry_computation()->instructions().end(), in TEST_F() 204 absl::c_copy_if(module->entry_computation()->instructions(), in TEST_F() 273 *(std::find_if(module->entry_computation()->instructions().begin(), in TEST_F() 274 module->entry_computation()->instructions().end(), in TEST_F() 282 HloInstruction* new_root = module->entry_computation()->root_instruction(); in TEST_F() [all …]
|
H A D | call_graph_test.cc | 119 HloComputation* entry_computation = in TEST_F() local 127 const CallGraphNode& entry_node = call_graph->GetNode(entry_computation); in TEST_F() 129 EXPECT_EQ(entry_computation, entry_node.computation()); in TEST_F() 145 HloComputation* entry_computation = module->AddEntryComputation( in TEST_F() local 151 const CallGraphNode& entry_node = call_graph->GetNode(entry_computation); in TEST_F() 152 EXPECT_EQ(entry_computation, entry_node.computation()); in TEST_F() 158 EXPECT_TRUE(call_graph->GetComputationCallers(entry_computation).empty()); in TEST_F() 178 HloComputation* entry_computation = module->AddEntryComputation( in TEST_F() local 188 const CallGraphNode& entry_node = call_graph->GetNode(entry_computation); in TEST_F() 189 EXPECT_EQ(entry_computation, entry_node.computation()); in TEST_F() [all …]
|
H A D | hlo_rematerialization_test.cc | 160 HloComputation* entry_computation = in TEST_F() local 164 EXPECT_EQ(entry_computation->instruction_count(), 7); in TEST_F() 176 EXPECT_EQ(entry_computation->instruction_count(), 8); in TEST_F() 196 HloComputation* entry_computation = in TEST_F() local 200 EXPECT_EQ(entry_computation->instruction_count(), 7); in TEST_F() 209 EXPECT_EQ(entry_computation->instruction_count(), 9); in TEST_F() 234 HloComputation* entry_computation = in TEST_F() local 238 EXPECT_EQ(entry_computation->instruction_count(), 7); in TEST_F() 250 EXPECT_EQ(entry_computation->instruction_count(), 9); in TEST_F() 292 HloComputation* entry_computation = in TEST_F() local [all …]
|
H A D | hlo_cost_analysis_test.cc | 149 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F() 160 HloInstruction* root = hlo_module->entry_computation()->root_instruction(); in TEST_F() 183 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F() 194 HloInstruction* root = hlo_module->entry_computation()->root_instruction(); in TEST_F() 219 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F() 230 HloInstruction* root = hlo_module->entry_computation()->root_instruction(); in TEST_F() 249 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F() 260 HloInstruction* root = hlo_module->entry_computation()->root_instruction(); in TEST_F() 276 hlo_module->entry_computation()->root_instruction()->Accept(&analysis)); in TEST_F() 283 HloInstruction* root = hlo_module->entry_computation()->root_instruction(); in TEST_F() [all …]
|
H A D | while_util_test.cc | 35 HloComputation** entry_computation, HloInstruction** param0, in GetParsedModule() argument 60 *entry_computation = module->entry_computation(); in GetParsedModule() 61 *param0 = (*entry_computation)->parameter_instruction(0); in GetParsedModule() 62 *param1 = (*entry_computation)->parameter_instruction(1); in GetParsedModule() 63 *param2 = (*entry_computation)->parameter_instruction(2); in GetParsedModule() 71 HloComputation* entry_computation; in TEST_F() local 75 GetParsedModule(&entry_computation, ¶m0, ¶m1, ¶m2)); in TEST_F() 77 HloInstruction* while_instr = entry_computation->root_instruction(); in TEST_F() 87 entry_computation->root_instruction(), in TEST_F() 102 HloComputation* entry_computation; in TEST_F() local [all …]
|
H A D | fusion_node_indexing_evaluation_test.cc | 89 HloInstruction* sub = module->entry_computation()->root_instruction(); in TEST_F() 91 InstructionFusionForTesting().Fuse(add, sub, module->entry_computation()); in TEST_F() 104 HloInstruction* sub = module->entry_computation()->root_instruction(); in TEST_F() 109 instruction_fusion.Fuse(slice1, sub, module->entry_computation()); in TEST_F() 111 instruction_fusion.Fuse(slice2, fusion, module->entry_computation()); in TEST_F() 145 HloInstruction* add2 = module->entry_computation()->root_instruction(); in TEST_F() 150 instruction_fusion.Fuse(slice2_0, add2, module->entry_computation()); in TEST_F() 155 instruction_fusion.Fuse(slice2_1, fusion, module->entry_computation()); in TEST_F() 160 instruction_fusion.Fuse(add1, fusion, module->entry_computation()); in TEST_F() 166 instruction_fusion.Fuse(slice1_0, fusion, module->entry_computation()); in TEST_F() [all …]
|
H A D | instruction_fusion_test.cc | 57 HloInstruction* sub = module->entry_computation()->root_instruction(); in TEST_F() 60 InstructionFusionForTesting().Fuse(add, sub, module->entry_computation()); in TEST_F() 81 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() 84 abs, root, module->entry_computation()); in TEST_F() 101 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() 105 abs, tanh, module->entry_computation()); in TEST_F() 196 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() 302 root = module->entry_computation()->root_instruction(); in TEST_F() 388 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() 417 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() [all …]
|
H A D | transpose_folding_test.cc | 63 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F() 105 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F() 160 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F() 182 HloComputation* entry_computation = in TEST_F() local 185 {add, sub, mul}, "entry", entry_computation); in TEST_F() 186 EXPECT_EQ(call, entry_computation->root_instruction()); in TEST_F() 260 HloComputation* entry_computation = in TEST_F() local 267 entry_computation->instructions().begin(), in TEST_F() 268 entry_computation->instructions().end()); in TEST_F() 318 HloComputation* entry_computation = in TEST_F() local [all …]
|
H A D | topk_rewriter_test.cc | 139 module->entry_computation()->root_instruction(), in TEST_F() 143 module->entry_computation()->root_instruction()->operand(0)->operand(0); in TEST_F() 170 module->entry_computation()->root_instruction(), in TEST_F() 174 module->entry_computation()->root_instruction()->operand(0)->operand(0); in TEST_F() 201 module->entry_computation()->root_instruction(), in TEST_F() 205 module->entry_computation()->root_instruction()->operand(0)->operand(0); in TEST_F() 231 LOG(INFO) << module->entry_computation()->ToString(); in TEST_F() 233 module->entry_computation()->root_instruction(), in TEST_F() 238 const HloInstruction* cc = module->entry_computation() in TEST_F() 262 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F() [all …]
|
H A D | while_loop_simplifier_test.cc | 39 const auto& instrs = m->entry_computation()->instructions(); in FindFirstWhile() 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() 167 HloComputation* computation = m->entry_computation(); in TEST_F() 184 HloComputation* computation = m->entry_computation(); in TEST_F() 200 HloComputation* computation = m->entry_computation(); in TEST_F() 216 HloComputation* computation = m->entry_computation(); in TEST_F() 231 HloComputation* computation = m->entry_computation(); in TEST_F() [all …]
|
H A D | conditional_code_motion_test.cc | 78 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() 163 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() 204 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() 289 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() 370 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() 438 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() 516 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() 722 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() 809 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() 854 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() [all …]
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
H A D | gpu_fusible_test.cc | 47 module->entry_computation()->root_instruction()->operand(0); in TEST_F() 79 module->entry_computation()->root_instruction()->operand(1); in TEST_F() 112 module->entry_computation()->root_instruction()->operand(1); in TEST_F() 132 module->entry_computation()->root_instruction()->operand(0); in TEST_F() 152 module->entry_computation()->root_instruction()->operand(0); in TEST_F() 183 module->entry_computation()->root_instruction()->operand(0); in TEST_F() 208 module->entry_computation()->root_instruction()->operand(0); in TEST_F() 224 module->entry_computation()->root_instruction(); in TEST_F() 242 module->entry_computation()->root_instruction(); in TEST_F() 261 module->entry_computation()->root_instruction(); in TEST_F() [all …]
|
H A D | gpu_conv_rewriter_test.cc | 126 HloComputation* entry_computation = in TEST_F() local 129 ASSERT_THAT(entry_computation->root_instruction(), in TEST_F() 135 entry_computation->root_instruction()->operand(0)->metadata(); in TEST_F() 163 HloComputation* entry_computation = in TEST_F() local 167 entry_computation->root_instruction(), in TEST_F() 193 HloComputation* entry_computation = in TEST_F() local 196 EXPECT_THAT(entry_computation->root_instruction(), in TEST_F() 223 HloComputation* entry_computation = in TEST_F() local 226 EXPECT_THAT(entry_computation->root_instruction(), in TEST_F() 252 HloComputation* entry_computation = in TEST_F() local [all …]
|
H A D | gpu_layout_assignment_test.cc | 110 module->entry_computation()->ComputeProgramShape(), in TEST_F() 115 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F() 136 module->entry_computation()->ComputeProgramShape(), in TEST_F() 142 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F() 163 module->entry_computation()->ComputeProgramShape(), in TEST_F() 169 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F() 189 module->entry_computation()->ComputeProgramShape(), in TEST_F() 195 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F() 216 module->entry_computation()->ComputeProgramShape(), in TEST_F() 222 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F() [all …]
|
H A 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() 94 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() 161 auto* root = module->entry_computation()->root_instruction(); in TEST_F() 182 auto* root = module->entry_computation()->root_instruction(); in TEST_F() 206 auto* root = module->entry_computation()->root_instruction(); in TEST_F() 230 auto* root = module->entry_computation()->root_instruction(); in TEST_F() 251 auto* root = module->entry_computation()->root_instruction(); in TEST_F() [all …]
|
H A D | cudnn_fused_conv_rewriter_test.cc | 454 EXPECT_THAT(m->entry_computation()->root_instruction(), in TEST_F() 491 m->entry_computation()->root_instruction(), in TEST_F() 531 m->entry_computation()->root_instruction(), in TEST_F() 581 m->entry_computation()->root_instruction(), in TEST_F() 629 m->entry_computation()->root_instruction(), in TEST_F() 674 m->entry_computation()->root_instruction(), in TEST_F() 712 m->entry_computation()->root_instruction(), in TEST_F() 751 m->entry_computation()->root_instruction(), in TEST_F() 793 m->entry_computation()->root_instruction(), in TEST_F() 831 m->entry_computation()->root_instruction(), in TEST_F() [all …]
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tools/ |
H A D | hlo_control_flow_flattening_test.cc | 70 auto root = module->entry_computation()->root_instruction(); in TEST_F() 71 auto while_op = module->entry_computation()->GetInstructionWithName("while"); in TEST_F() 131 auto root = module->entry_computation()->root_instruction(); in TEST_F() 132 auto while_op = module->entry_computation()->GetInstructionWithName("while"); in TEST_F() 185 auto root = module->entry_computation()->root_instruction(); in TEST_F() 186 auto while_op = module->entry_computation()->GetInstructionWithName("while"); in TEST_F() 241 auto fusion = module->entry_computation()->root_instruction(); in TEST_F() 242 auto while_op = module->entry_computation()->GetInstructionWithName("while"); in TEST_F() 264 auto tuple = module->entry_computation()->root_instruction(); in TEST_F() 278 Shape root_shape = module->entry_computation()->root_instruction()->shape(); in TEST_F() [all …]
|
H A D | hlo_extractor_test.cc | 47 EXPECT_THAT(extracted_module->entry_computation()->root_instruction(), in TEST_F() 54 EXPECT_THAT(extracted_module->entry_computation()->root_instruction(), in TEST_F() 61 EXPECT_THAT(extracted_module->entry_computation()->root_instruction(), in TEST_F() 85 EXPECT_THAT(extracted_module->entry_computation()->root_instruction(), in TEST_F() 92 EXPECT_THAT(extracted_module->entry_computation()->root_instruction(), in TEST_F() 98 EXPECT_THAT(extracted_module->entry_computation()->root_instruction(), in TEST_F() 105 EXPECT_THAT(extracted_module->entry_computation()->root_instruction(), in TEST_F() 129 EXPECT_THAT(extracted_module->entry_computation()->root_instruction(), in TEST_F() 135 EXPECT_THAT(extracted_module->entry_computation()->root_instruction(), in TEST_F()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
H A D | xla_builder_test.cc | 77 auto root = module->entry_computation()->root_instruction(); in TEST_F() 88 auto root = module->entry_computation()->root_instruction(); in TEST_F() 102 auto root = module->entry_computation()->root_instruction(); in TEST_F() 133 auto root = module->entry_computation()->root_instruction(); in TEST_F() 150 module->entry_computation()->root_instruction(), in TEST_F() 166 module->entry_computation()->root_instruction(), in TEST_F() 188 auto root = module->entry_computation()->root_instruction(); in TEST_F() 204 auto root = module->entry_computation()->root_instruction(); in TEST_F() 213 auto root = module->entry_computation()->root_instruction(); in TEST_F() 265 auto root = module->entry_computation()->root_instruction(); in TEST_F() [all …]
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
H A D | spmd_partitioner_test.cc | 110 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() 133 EXPECT_NE(custom_call->parent(), module->entry_computation()); in TEST_F() 134 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() 152 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() 173 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() 197 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() 219 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() 242 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() 262 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() 286 HloInstruction* root = module->entry_computation()->root_instruction(); in TEST_F() [all …]
|
H A D | schedule_aware_collective_ops_cse_test.cc | 60 HloInstruction* tuple = module->entry_computation()->root_instruction(); in TEST_F() 81 HloInstruction* tuple = module->entry_computation()->root_instruction(); in TEST_F() 104 HloInstruction* tuple = module->entry_computation()->root_instruction(); in TEST_F() 127 HloInstruction* tuple = module->entry_computation()->root_instruction(); in TEST_F() 150 HloInstruction* tuple = module->entry_computation()->root_instruction(); in TEST_F() 171 HloInstruction* tuple = module->entry_computation()->root_instruction(); in TEST_F() 194 HloInstruction* tuple = module->entry_computation()->root_instruction(); in TEST_F() 215 HloInstruction* tuple = module->entry_computation()->root_instruction(); in TEST_F() 236 HloInstruction* tuple = module->entry_computation()->root_instruction(); in TEST_F()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
H A D | gpu_reduce_scatter_creator_test.cc | 58 return absl::c_count_if(module->entry_computation()->instructions(), in AllReduceCount() 95 ASSERT_THAT(module->entry_computation()->root_instruction(), in TEST_F() 98 module->entry_computation()->root_instruction()); in TEST_F() 134 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F() 167 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F() 201 ASSERT_THAT(module->entry_computation()->root_instruction(), in TEST_F() 204 module->entry_computation()->root_instruction()); in TEST_F() 269 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F() 304 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F() 338 EXPECT_THAT(module->entry_computation()->root_instruction(), in TEST_F() [all …]
|