Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_creation_utils_test.cc35 HloComputation** entry_computation) { in CreateModuleWithProgramShape()
50 HloComputation** entry_computation, PrimitiveType primitive_type_output) { in CreateModuleWithProgramShape()
65 HloComputation* entry_computation; in TEST_F() local
84 HloComputation* entry_computation; in TEST_F() local
107 HloComputation* entry_computation; in TEST_F() local
126 HloComputation* entry_computation; in TEST_F() local
145 HloComputation* entry_computation; in TEST_F() local
164 HloComputation* entry_computation; in TEST_F() local
185 HloComputation* entry_computation; in TEST_F() local
205 HloComputation* entry_computation; in TEST_F() local
[all …]
H A Dcall_graph_test.cc119 HloComputation* entry_computation = in TEST_F() local
145 HloComputation* entry_computation = module->AddEntryComputation( in TEST_F() local
178 HloComputation* entry_computation = module->AddEntryComputation( in TEST_F() local
221 HloComputation* entry_computation = in TEST_F() local
271 HloComputation* entry_computation = in TEST_F() local
336 HloComputation* entry_computation; in TEST_F() local
469 HloComputation* entry_computation; in TEST_F() local
523 HloComputation* entry_computation = in TEST_F() local
H A 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
H A Dtranspose_folding_test.cc182 HloComputation* entry_computation = in TEST_F() local
260 HloComputation* entry_computation = in TEST_F() local
318 HloComputation* entry_computation = in TEST_F() local
381 HloComputation* entry_computation = in TEST_F() local
450 HloComputation* entry_computation = in TEST_F() local
H A 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
H A 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
H A Dflatten_call_graph_test.cc132 HloComputation* entry_computation; in TEST_F() local
168 HloComputation* entry_computation; in TEST_F() local
H A Dhlo_profile_printer_data.proto66 string entry_computation = 4; field
H A Dhlo_dce.cc138 if (HloComputation* entry_computation = module->entry_computation()) { in RecursivelyRemoveDeadComputations() local
H A Dheap_simulator_test.cc82 HloComputation* entry_computation = in TEST_F() local
190 auto entry_computation = module->AddEntryComputation(builder.Build()); in TEST_F() local
286 std::unique_ptr<HloComputation> entry_computation, in HeapSimulatorTracker()
H A Dmemory_space_assignment_test.cc203 const HloComputation* entry_computation = module->entry_computation(); in CheckParametersInDefaultMemory() local
833 HloComputation* entry_computation = in TEST_P() local
2648 HloComputation* entry_computation = in TEST_P() local
2721 HloComputation* entry_computation = in TEST_P() local
2788 HloComputation* entry_computation = in TEST_P() local
2860 HloComputation* entry_computation = in TEST_P() local
2977 HloComputation* entry_computation = in TEST_P() local
3080 HloComputation* entry_computation = in TEST_P() local
H A Dhlo_live_range_test.cc320 HloComputation* entry_computation = in TEST_F() local
H A Dhlo_module.h126 HloComputation* entry_computation() const { in entry_computation() function
H A Dhlo_alias_analysis.cc53 const HloComputation& entry_computation = *module.entry_computation(); in ComputeInputOutputAliasedValues() local
H A Dhlo_module.cc67 void HloModule::ReplaceEntryComputation(HloComputation* entry_computation) { in ReplaceEntryComputation()
H A Dheap_simulator.cc105 const HloComputation* entry_computation = module.entry_computation(); in Run() local
H A Dmemory_space_assignment.cc2400 HloComputation* entry_computation = module.entry_computation(); in AddInputAndOutputRequiredAssignments() local
2519 const HloComputation* entry_computation = module.entry_computation(); in AreIntervalsReservedInAlternateMemory() local
H A Dhlo_parser.cc978 HloComputation* entry_computation = nullptr; in ParseComputations() local
1015 bool HloParserImpl::ParseComputation(HloComputation** entry_computation) { in ParseComputation()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_conv_rewriter_test.cc126 HloComputation* entry_computation = in TEST_F() local
163 HloComputation* entry_computation = in TEST_F() local
193 HloComputation* entry_computation = in TEST_F() local
223 HloComputation* entry_computation = in TEST_F() local
252 HloComputation* entry_computation = in TEST_F() local
306 HloComputation* entry_computation = in TEST_F() local
356 HloComputation* entry_computation = in TEST_F() local
391 HloComputation* entry_computation = in TEST_F() local
446 HloComputation* entry_computation = in TEST_F() local
497 HloComputation* entry_computation = in TEST_F() local
[all …]
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dir_emission_utils_test.cc44 HloComputation* entry_computation = module->entry_computation(); in TEST_F() local
H A Dconv_canonicalization_test.cc91 HloComputation* entry_computation = in TEST_F() local
H A Dcpu_compiler.cc791 const HloComputation& entry_computation = *module.entry_computation(); in CreateHloProfilingArtifacts() local
1146 HloComputation* entry_computation = module->entry_computation(); in CompileLegacyCpuExecutable() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
H A Dcpu_eigen_dot_operation_test.cc47 void CompileAndCheck(std::unique_ptr<HloComputation> entry_computation, in CompileAndCheck()