Home
last modified time | relevance | path

Searched defs:entry_computation (Results 1 – 25 of 25) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_creation_utils_test.cc33 HloComputation** entry_computation) { in CreateModuleWithProgramShape()
48 HloComputation** entry_computation, PrimitiveType primitive_type_output) { in CreateModuleWithProgramShape()
63 HloComputation* entry_computation; in TEST_F() local
82 HloComputation* entry_computation; in TEST_F() local
105 HloComputation* entry_computation; in TEST_F() local
124 HloComputation* entry_computation; in TEST_F() local
143 HloComputation* entry_computation; in TEST_F() local
162 HloComputation* entry_computation; in TEST_F() local
183 HloComputation* entry_computation; in TEST_F() local
203 HloComputation* entry_computation; in TEST_F() local
[all …]
Dtuple_util_test.cc34 HloComputation** entry_computation, HloInstruction** param0, in GetParsedModule()
60 HloComputation* entry_computation; in TEST() local
73 HloComputation* entry_computation; in TEST() local
Dcall_graph_test.cc118 HloComputation* entry_computation = in TEST_F() local
144 HloComputation* entry_computation = module->AddEntryComputation( in TEST_F() local
175 HloComputation* entry_computation = module->AddEntryComputation( in TEST_F() local
216 HloComputation* entry_computation = in TEST_F() local
266 HloComputation* entry_computation = in TEST_F() local
331 HloComputation* entry_computation; in TEST_F() local
465 HloComputation* entry_computation; in TEST_F() local
519 HloComputation* entry_computation = in TEST_F() local
Dwhile_util_test.cc35 HloComputation** entry_computation, HloInstruction** param0, in GetParsedModule()
71 HloComputation* entry_computation; in TEST_F() local
102 HloComputation* entry_computation; in TEST_F() local
Dhlo_rematerialization_test.cc160 HloComputation* entry_computation = in TEST_F() local
196 HloComputation* entry_computation = in TEST_F() local
234 HloComputation* entry_computation = in TEST_F() local
292 HloComputation* entry_computation = in TEST_F() local
380 HloComputation* entry_computation = in TEST_F() local
449 HloComputation* entry_computation = in TEST_F() local
535 HloComputation* entry_computation = in TEST_P() local
Dtranspose_folding_test.cc177 HloComputation* entry_computation = in TEST_F() local
254 HloComputation* entry_computation = in TEST_F() local
311 HloComputation* entry_computation = in TEST_F() local
373 HloComputation* entry_computation = in TEST_F() local
441 HloComputation* entry_computation = in TEST_F() local
Dhlo_dce.cc92 if (HloComputation* entry_computation = module->entry_computation()) { in Run() local
Dhlo_profile_printer_data.proto66 string entry_computation = 4; field
Dflatten_call_graph_test.cc132 HloComputation* entry_computation; in TEST_F() local
168 HloComputation* entry_computation; in TEST_F() local
Dwhile_loop_invariant_code_motion_test.cc89 HloComputation* entry_computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
148 HloComputation* entry_computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
388 HloComputation* entry_computation = m->AddEntryComputation(builder.Build()); in TEST_F() local
Dhlo_module.h123 HloComputation* entry_computation() const { in entry_computation() function
Dhlo_live_range_test.cc320 HloComputation* entry_computation = in TEST_F() local
Dheap_simulator_test.cc83 HloComputation* entry_computation = in TEST_F() local
191 auto entry_computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
286 const string& name, std::unique_ptr<HloComputation> entry_computation, in HeapSimulatorTracker()
Dmemory_space_assignment_test.cc197 const HloComputation* entry_computation = module->entry_computation(); in CheckParametersInDefaultMemory() local
817 HloComputation* entry_computation = in TEST_P() local
2706 HloComputation* entry_computation = in TEST_P() local
2779 HloComputation* entry_computation = in TEST_P() local
2846 HloComputation* entry_computation = in TEST_P() local
2918 HloComputation* entry_computation = in TEST_P() local
3035 HloComputation* entry_computation = in TEST_P() local
3138 HloComputation* entry_computation = in TEST_P() local
Dhlo_module.cc61 void HloModule::ReplaceEntryComputation(HloComputation* entry_computation) { in ReplaceEntryComputation()
Dheap_simulator.cc102 const HloComputation* entry_computation = module.entry_computation(); in Run() local
Dmemory_space_assignment.cc1923 HloComputation* entry_computation = module.entry_computation(); in AddInputAndOutputRequiredAssignments() local
2010 const HloComputation* entry_computation = module.entry_computation(); in AreIntervalsReservedInAlternateMemory() local
Dhlo_parser.cc893 HloComputation* entry_computation = nullptr; in ParseComputations() local
929 bool HloParserImpl::ParseComputation(HloComputation** entry_computation) { in ParseComputation()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_rewriter_test.cc125 HloComputation* entry_computation = in TEST_F() local
162 HloComputation* entry_computation = in TEST_F() local
192 HloComputation* entry_computation = in TEST_F() local
222 HloComputation* entry_computation = in TEST_F() local
251 HloComputation* entry_computation = in TEST_F() local
305 HloComputation* entry_computation = in TEST_F() local
355 HloComputation* entry_computation = in TEST_F() local
390 HloComputation* entry_computation = in TEST_F() local
445 HloComputation* entry_computation = in TEST_F() local
496 HloComputation* entry_computation = in TEST_F() local
[all …]
Dvariadic_op_splitter_test.cc58 auto entry_computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
73 auto entry_computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
Dgpu_hlo_schedule.cc320 HloComputation* entry_computation = module->entry_computation(); in Build() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emission_utils_test.cc44 HloComputation* entry_computation = module->entry_computation(); in TEST_F() local
Dconv_canonicalization_test.cc91 HloComputation* entry_computation = in TEST_F() local
Dcpu_compiler.cc571 const HloComputation& entry_computation = *module.entry_computation(); in CreateHloProfilingArtifacts() local
720 HloComputation* entry_computation = module->entry_computation(); in RunBackend() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_eigen_dot_operation_test.cc46 void CompileAndCheck(std::unique_ptr<HloComputation> entry_computation, in CompileAndCheck()