Home
last modified time | relevance | path

Searched refs:subcomputation (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_copy_insertion_test.cc122 HloComputation* subcomputation = in TEST_F() local
126 HloInstruction::CreateCall(tuple_shape, {param}, subcomputation)); in TEST_F()
132 EXPECT_EQ(CountCopies(*subcomputation), 2); in TEST_F()
133 EXPECT_THAT(subcomputation->root_instruction(), in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_rematerialization_test.cc425 HloComputation* subcomputation = nullptr; in TEST_F() local
436 subcomputation = module->AddEmbeddedComputation(builder.Build()); in TEST_F()
447 HloInstruction::CreateCall(vec1024_shape_, {add_1}, subcomputation)); in TEST_F()
451 HloInstruction::CreateCall(vec1024_shape_, {add_2}, subcomputation)); in TEST_F()
455 HloInstruction::CreateCall(vec1024_shape_, {add_3}, subcomputation)); in TEST_F()
584 HloComputation* subcomputation = nullptr; in TEST_P() local
595 subcomputation = module->AddEmbeddedComputation(builder.Build()); in TEST_P()
606 HloInstruction::CreateCall(vec1024_shape_, {add_1}, subcomputation)); in TEST_P()
Dcall_graph_test.cc206 HloComputation* subcomputation = in TEST_F() local
213 HloInstruction::CreateCall(kScalarShape, {param0}, subcomputation)); in TEST_F()
215 HloInstruction::CreateMap(kScalarShape, {call}, subcomputation)); in TEST_F()
231 UnorderedElementsAre(subcomputation)); in TEST_F()
238 UnorderedElementsAre(subcomputation)); in TEST_F()
242 const CallGraphNode& sub_node = call_graph->GetNode(subcomputation); in TEST_F()
Dhlo_ordering_test.cc468 HloComputation* subcomputation = module->AddEmbeddedComputation( in TEST_F() local
475 HloInstruction::CreateCall(scalar_shape, {c}, subcomputation)); in TEST_F()
479 schedule.set_sequence(subcomputation, {param, root, dead}); in TEST_F()
Dbuffer_assignment.cc170 for (HloComputation* subcomputation : in GatherComputationsByAllocationType()
185 worklist.push_back(std::make_pair(subcomputation, in GatherComputationsByAllocationType()
197 worklist.push_back(std::make_pair(subcomputation, in GatherComputationsByAllocationType()
/external/jline/src/src/test/resources/jline/example/
Denglish.gz