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_compiler.cc1224 for (ComputationToEmit subcomputation : in CompileLegacyCpuExecutable() local
1226 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() local
1502 if (subcomputation.computation->IsFusionComputation()) { in CompileAheadOfTime()
1507 .EmitComputation(subcomputation.computation, in CompileAheadOfTime()
1508 subcomputation.computation->name(), in CompileAheadOfTime()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_rematerialization_test.cc347 HloComputation* subcomputation = nullptr; in TEST_F() local
358 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() local
515 subcomputation = module->AddEmbeddedComputation(builder.Build()); in TEST_P()
526 HloInstruction::CreateCall(vec1024_shape_, {add_1}, subcomputation)); in TEST_P()
Dcall_graph_test.cc211 HloComputation* subcomputation = in TEST_F() local
218 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()
Dhlo_ordering_test.cc475 HloComputation* subcomputation = module->AddEmbeddedComputation( in TEST_F() local
482 HloInstruction::CreateCall(scalar_shape, {c}, subcomputation)); in TEST_F()
486 schedule.set_sequence(subcomputation, {param, root, dead}); in TEST_F()
Dbuffer_assignment.cc154 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()
/external/jline/src/src/test/resources/jline/example/
Denglish.gz