Searched refs:subcomputation (Results 1 – 9 of 9) sorted by relevance
1224 for (ComputationToEmit subcomputation : in CompileLegacyCpuExecutable() local1226 if (subcomputation.computation->IsFusionComputation()) { in CompileLegacyCpuExecutable()1232 subcomputation.computation, in CompileLegacyCpuExecutable()1233 subcomputation.computation->name(), in CompileLegacyCpuExecutable()1235 schedule.sequence(subcomputation.computation).instructions(), in CompileLegacyCpuExecutable()1236 subcomputation.allow_reassociation) in CompileLegacyCpuExecutable()1500 for (ComputationToEmit subcomputation : in CompileAheadOfTime() local1502 if (subcomputation.computation->IsFusionComputation()) { in CompileAheadOfTime()1507 .EmitComputation(subcomputation.computation, in CompileAheadOfTime()1508 subcomputation.computation->name(), in CompileAheadOfTime()[all …]
347 HloComputation* subcomputation = nullptr; in TEST_F() local358 subcomputation = module->AddEmbeddedComputation(builder.Build()); in TEST_F()369 HloInstruction::CreateCall(vec1024_shape_, {add_1}, subcomputation)); in TEST_F()373 HloInstruction::CreateCall(vec1024_shape_, {add_2}, subcomputation)); in TEST_F()377 HloInstruction::CreateCall(vec1024_shape_, {add_3}, subcomputation)); in TEST_F()504 HloComputation* subcomputation = nullptr; in TEST_P() local515 subcomputation = module->AddEmbeddedComputation(builder.Build()); in TEST_P()526 HloInstruction::CreateCall(vec1024_shape_, {add_1}, subcomputation)); in TEST_P()
211 HloComputation* subcomputation = in TEST_F() local218 HloInstruction::CreateCall(kScalarShape, {param0}, subcomputation)); in TEST_F()220 HloInstruction::CreateMap(kScalarShape, {call}, subcomputation)); in TEST_F()236 UnorderedElementsAre(subcomputation)); in TEST_F()243 UnorderedElementsAre(subcomputation)); in TEST_F()247 const CallGraphNode& sub_node = call_graph->GetNode(subcomputation); in TEST_F()
475 HloComputation* subcomputation = module->AddEmbeddedComputation( in TEST_F() local482 HloInstruction::CreateCall(scalar_shape, {c}, subcomputation)); in TEST_F()486 schedule.set_sequence(subcomputation, {param, root, dead}); in TEST_F()
154 for (HloComputation* subcomputation : in GatherComputationsByAllocationType()173 worklist.push_back(std::make_pair(subcomputation, in GatherComputationsByAllocationType()188 worklist.push_back(std::make_pair(subcomputation, in GatherComputationsByAllocationType()
1Aarhus 2Aaron 3Ababa 4aback 5abaft 6abandon 7abandoned 8abandoning 9abandonment 10 ...
1dictionary=main:en_gb,locale=en_GB,description=English (UK),date ...
1dictionary=main:en,locale=en,description=English,date=1414726273, ...