| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tools/ |
| H A D | interactive_graphviz.cc | 128 for (const auto& computation : module.computations()) { in FindInstruction() local
144 for (auto* computation : module.computations()) { in FindComputation() local
236 for (const auto& computation : module.computations()) { in DoListComputationsCommand() local
261 for (const auto& computation : module.computations()) { in DoListCommand() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | parallel_task_assignment.cc | 120 HloComputation* computation = module->entry_computation(); in ParallelTaskAssignment() local
200 HloModule* module, HloComputation* computation, in AssignParallelTasksHelper()
260 for (auto* computation : module->MakeNonfusionComputations()) { in ComputeTargetParallelTasks() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_module_dce_test.cc | 37 bool HasInstruction(const HloComputation& computation, in HasInstruction()
45 bool WhileBodyHasPassThroughTupleElement(const HloComputation* computation, in WhileBodyHasPassThroughTupleElement()
71 const HloComputation* computation) { in GetWhileLoops()
|
| H A D | heap_simulator.cc | 73 const HloComputation& computation, const HloInstructionSequence& sequence, in MinimumMemoryForComputation()
87 const HloComputation& computation, const HloInstructionSequence& sequence, in MinimumMemoryForComputation()
120 const HloComputation& computation, in Run()
141 const HloComputation& computation, in Run()
159 const HloComputation& computation, in RunComputation()
|
| H A D | hlo_live_range.cc | 35 const HloComputation* computation, bool module_scoped_analysis) { in Run()
72 void HloLiveRange::FlattenSchedule(const HloComputation& computation, in FlattenSchedule()
181 const HloComputation* computation = instruction.parent(); in CalculateBufferStartEndMap() local
|
| H A D | gather_expander.cc | 115 HloComputation* computation = index_vector->parent(); in ExpandIndexVectorIntoOperandSpace() local
235 HloComputation* computation, PrimitiveType element_type, in CreateGatherLoopAccumulatorInitValue()
353 HloComputation* computation = gather_instr->parent(); in ExpandInstruction() local
|
| H A D | hlo_module_test.cc | 69 for (auto computation : computations) { in CreateCallComputation() local
82 auto computation = module->AddEntryComputation(CreateConstantComputation()); in TEST_F() local
467 for (const HloComputation* computation : module_copy->computations()) { in TEST_F() local
528 auto computation = module->AddEntryComputation(CreateConstantComputation()); in TEST_F() local
|
| H A D | sharding_propagation.cc | 2288 for (auto computation : module->computations(execution_threads)) { in Run() local
2298 for (auto computation : module->computations(execution_threads)) { in Run() local
2382 for (auto computation : module->computations(execution_threads)) { in Run() local
2393 for (auto computation : module->computations(execution_threads)) { in Run() local
2430 for (const HloComputation* computation : in Run() local
2460 for (const HloComputation* computation : in Run() local
2594 for (auto computation : module->computations(execution_threads)) { in Run() local
|
| H A D | hlo_creation_utils.h | 245 StatusOr<HloInstruction*> MakeR1ConstantHlo(HloComputation* computation, in MakeR1ConstantHlo()
259 HloInstruction* MakeR0ConstantHlo(HloComputation* computation, NativeT value) { in MakeR0ConstantHlo()
|
| H A D | hlo_rematerialization_test_utils_test.cc | 39 auto computation = MakeRematerializableComputation(); in TEST_F() local
72 auto computation = MakeRematerializableWhileComputation( in TEST_F() local
|
| H A D | root_instruction_sinker.cc | 24 void SinkTupleRoot(HloComputation* computation) { in SinkTupleRoot()
39 void SinkNontupleRoot(HloComputation* computation) { in SinkNontupleRoot()
|
| H A D | logistic_expander_test.cc | 63 auto computation = m->entry_computation(); in TEST_F() local
90 auto computation = m->entry_computation(); in TEST_F() local
|
| H A D | all_reduce_simplifier.cc | 57 for (auto computation : module->computations(execution_threads)) { in Run() local
70 for (auto computation : module->computations(execution_threads)) { in Run() local
|
| H A D | hlo_instruction.cc | 1859 for (const auto& computation : called_computations()) { in HasSideEffect() local
1874 HloComputation* computation) { in CreateCall()
2806 void HloInstruction::set_to_apply(HloComputation* computation) { in set_to_apply()
2839 void HloInstruction::set_while_condition(HloComputation* computation) { in set_while_condition()
2847 void HloInstruction::set_while_body(HloComputation* computation) { in set_while_body()
2891 HloComputation* computation) { in set_branch_computation()
3225 [&](std::string* out, const HloComputation* computation) { in ExtraAttributesToString()
3248 [&](std::string* out, const HloComputation* computation) { in ExtraAttributesToString()
3264 [&](std::string* out, const HloComputation* computation) { in ExtraAttributesToString()
3296 [&](std::string* out, const HloComputation* computation) { in ExtraAttributesToString()
[all …]
|
| H A D | hlo_rematerialization.cc | 606 const HloComputation* computation() const { return computation_; } in computation() function in xla::__anon2b46f4430111::MemoryUsageTracker
793 const HloComputation* computation, in MemoryUsageTracker()
1563 HloComputation* computation = best->parent(); in RematerializeInstructions() local
1734 HloComputation* computation = best->parent(); in CompressInstruction() local
1844 const HloComputation* computation, const HloInstructionSequence& order, in ComputePeakMemory()
1876 for (const HloComputation* computation : callsite->called_computations()) { in CalledComputationsMemoryUsage() local
1894 HloComputation* computation, HloSchedule* schedule, in RematerializeComputation()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| H A D | utils.cc | 77 const XlaComputation& computation, const ProgramShape& program_shape) { in GetShardedProgramShapes()
149 const XlaComputation& computation, in DetermineArgumentLayoutsFromCompileOptions()
213 HloComputation* computation = module.entry_computation(); in ComputeParametersThatMustBeDonated() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/ |
| H A D | xla_compiler.cc | 82 const XlaComputation& computation) { in GetComputationSerializedProto()
114 const XlaComputation& computation) { in GetHloModule()
126 const XlaComputation& computation, bool print_large_constants = false) { in GetComputationHloText()
137 const XlaComputation& computation) { in GetComputationHloDotGraph()
146 StatusOr<uint64_t> HashComputation(const XlaComputation& computation) { in HashComputation()
|
| H A D | mlir.cc | 42 const XlaComputation& computation) { in PyXlaComputationToMlirModule()
78 XlaComputation computation; in PyMlirModuleToXlaComputation() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | client_library_test_base.cc | 112 const XlaComputation& computation, absl::Span<GlobalData* const> arguments, in ExecuteAndTransfer()
132 const XlaComputation& computation, absl::Span<GlobalData* const> arguments, in ExecuteAndTransferReference()
150 auto computation = std::move(computation_status).value(); in ExecuteToString() local
185 const xla::XlaComputation& computation, const Literal& expected, in ComputeAndCompareLiteralWithAllOutputLayouts()
211 const xla::XlaComputation& computation, const Literal& /*expected*/, in ComputeAndCompareLiteralWithAllInputLayouts()
|
| H A D | bad_rng_shape_validation_test.cc | 43 StatusOr<XlaComputation> computation = builder.Build(); in TEST_F() local
60 StatusOr<XlaComputation> computation = builder.Build(); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| H A D | cpu_fusion_test.cc | 74 auto computation = module->entry_computation(); in TEST_F() local
121 auto computation = module->entry_computation(); in TEST_F() local
196 auto computation = module->entry_computation(); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | tpu_strategy_compilation_test.py | 60 def computation(): function
68 def computation(): function
|
| /aosp_15_r20/external/aws-sdk-java-v2/core/http-auth-aws/src/main/java/software/amazon/awssdk/http/auth/aws/internal/signer/ |
| H A D | PrecomputedSha256Checksummer.java | 36 private final Callable<String> computation; field in PrecomputedSha256Checksummer
38 public PrecomputedSha256Checksummer(Callable<String> computation) { in PrecomputedSha256Checksummer()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| H A D | mlir_hlo_builder.cc | 144 const XlaComputation* computation, const Shape& shape, in CustomCallInternal()
226 const XlaComputation& computation, in ReduceInternal()
244 const XlaComputation& computation, Window window) { in ReduceWindowInternal()
737 Status MlirHloBuilder::ImportComputation(const HloModuleProto& computation, in ImportComputation()
747 Status MlirHloBuilder::ImportComputation(const HloModuleProto& computation, in ImportComputation()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/ |
| H A D | xla.proto | 577 HloModuleProto computation = 1; field
601 HloModuleProto computation = 1; field
628 HloModuleProto computation = 1; field
658 HloModuleProto computation = 1; field
|