| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_parser.cc | 996 auto computation = module->AddEntryComputation(std::move(computations_[i])); in ParseComputations() local
1031 HloComputation* computation = nullptr; in ParseComputation() local
1067 bool HloParserImpl::ParseInstructionList(HloComputation** computation, in ParseInstructionList()
4541 std::pair<HloComputation*, LocTy>* computation = in ParseComputationName() local
4858 HloComputation* computation; in ParseHloComputationList() local
5761 HloComputation* computation, in AddComputation()
|
| H A D | sort_simplifier.cc | 38 HloComputation* computation = sort->parent(); in RemoveUnusedOperandFromSort() local
|
| H A D | while_loop_constant_sinking.cc | 53 HloComputation* computation) { in CloneHelper()
|
| H A D | tuple_points_to_analysis.cc | 158 for (auto* computation : module_->MakeNonfusionComputations()) { in Analyze() local
670 for (const auto* computation : module_->MakeNonfusionComputations()) { in ToString() local
|
| H A D | stable_sort_expander.cc | 40 HloComputation* computation = sort->parent(); in ExpandInstruction() local
|
| H A D | convolution_4d_expander.cc | 58 HloComputation* computation = instruction->parent(); in ExpandInstruction() local
|
| H A D | multi_output_fusion.h | 101 HloComputation* computation() const { return computation_; } in computation() function
|
| H A D | all_reduce_folder.cc | 151 for (auto computation : module->computations(execution_threads)) { in Run() local
|
| H A D | hlo_execution_profile.cc | 86 const HloComputation* computation = pair.first; in CreateHloProfilePrinterData() local
|
| H A D | slice_sinker.cc | 173 HloComputation* computation = operation_on_slices[0]->parent(); in SinkSlices() local
|
| H A D | hlo_module_group_metadata.h | 246 const HloComputation* computation) const { in GetTrackedInstruction()
|
| H A D | fusion_node_indexing_evaluation_test.cc | 57 HloComputation* computation) override { in Fuse()
|
| H A D | copy_insertion_test.cc | 42 int64_t CountCopies(const HloComputation& computation) { in CountCopies()
54 for (const auto& computation : module.computations()) { in CountCopies() local
60 int64_t CountControlEdges(const HloComputation& computation) { in CountControlEdges()
70 for (const auto& computation : module.computations()) { in CountControlEdges() local
|
| H A D | hlo_sharding_util.cc | 270 void AssignComputationDevice(HloComputation* computation, int64_t device) { in AssignComputationDevice()
573 for (const HloComputation* computation : module.computations()) { in ContainsTileSharding() local
1174 auto computation = scatter.to_apply(); in IdentityValueAndHloOpcodeForScatterReduceComputation() local
|
| H A D | select_and_scatter_expander.cc | 32 auto* computation = instruction->parent(); in ExpandInstruction() local
|
| H A D | dump.cc | 371 static bool IsTrivial(const HloComputation& computation) { in IsTrivial()
447 for (const HloComputation* computation : in DumpHloModuleImpl() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | cpu_layout_assignment.cc | 114 const HloComputation* computation = constraints->computation(); in AddBackendConstraints() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| H A D | cpu_vectorization_test.cc | 99 std::unique_ptr<HloComputation> computation = builder.Build(); in TEST_P() local
|
| H A D | cpu_intrinsic_test.cc | 101 std::unique_ptr<HloComputation> computation = builder.Build(); in TEST_P() local
|
| H A D | cpu_noalias_test.cc | 58 std::unique_ptr<HloComputation> computation = builder.Build(); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | mlir_tf2xla.cc | 101 xla::XlaComputation* computation, absl::string_view debug_info_filename, in ConvertGraphDefToXlaViaMlir()
|
| H A D | xla_helpers.h | 161 std::shared_ptr<xla::XlaComputation> computation; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tools/ |
| H A D | hlo_extractor.cc | 87 for (auto computation : old_module_.MakeComputationPostOrder()) { in FinishVisit() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | deallocation_test.cc | 40 XlaComputation computation = builder->Build().value(); in ExecuteAndCheckTransfer() local
|
| /aosp_15_r20/development/tools/winscope/src/parsers/surface_flinger/computations/ |
| H A D | visibility_properties_computation_test.ts | 29 let computation: VisibilityPropertiesComputation; variable
|