Home
last modified time | relevance | path

Searched refs:computations (Results 1 – 25 of 214) sorted by relevance

123456789

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_subcomputation_unification.cc26 const auto& computations = module->computations(); in Run() local
27 for (auto i = computations.begin(); i != computations.end(); ++i) { in Run()
28 for (auto j = computations.begin(); j != i; ++j) { in Run()
Dhlo_module_group_util.h59 absl::Span<HloComputation* const> computations);
96 Status VerifyComputations(absl::Span<HloComputation* const> computations);
103 absl::Span<HloComputation* const> computations);
Dhlo_module_group_util.cc196 absl::Span<HloComputation* const> computations) { in RootInstructions() argument
198 for (HloComputation* computation : computations) { in RootInstructions()
318 absl::Span<HloComputation* const> computations) { in VerifyComputations() argument
326 for (HloComputation* computation : computations) { in VerifyComputations()
349 absl::Span<HloComputation* const> computations) { in ComputeReachability() argument
359 for (HloInstruction* root : RootInstructions(computations)) { in ComputeReachability()
Dhlo_module.cc250 for (const HloComputation* computation : computations()) { in CheckUniqueNamesAndIdsForComputationsAndInstructions()
307 std::vector<std::unique_ptr<HloComputation>> computations; in CreateFromProto() local
309 for (const HloComputationProto& computation_proto : proto.computations()) { in CreateFromProto()
322 computations.push_back(std::move(computation)); in CreateFromProto()
329 absl::c_sort(computations, [&](const std::unique_ptr<HloComputation>& a, in CreateFromProto()
335 for (auto& computation : computations) { in CreateFromProto()
550 for (auto* c : computations()) { in MakeNonfusionComputations()
575 for (HloComputation* computation : computations()) { in Clone()
612 auto computations_in_module = computations(); in GetComputationWithName()
Dhlo_module_test.cc52 absl::Span<HloComputation* const> computations) { in CreateCallComputation() argument
54 for (auto computation : computations) { in CreateCallComputation()
316 auto computation_copy_it = module_copy->computations().begin(); in TEST_F()
317 for (const HloComputation* computation_orig : module->computations()) { in TEST_F()
342 for (const HloComputation* computation : module_copy->computations()) { in TEST_F()
Dcall_graph.cc197 for (const HloComputation* computation : module_->computations()) { in SetCallContexts()
235 for (const HloComputation* computation : module_->computations()) { in SetCallContexts()
250 for (const HloComputation* computation : module_->computations()) { in SetNodeDepths()
284 for (HloComputation* computation : module->computations()) { in Build()
299 for (const HloComputation* computation : module->computations()) { in Build()
Dhlo_pass_pipeline_test.cc49 for (HloComputation* computation : module->computations()) { in Run()
68 for (HloComputation* computation : module->computations()) { in RunOnModuleGroup()
87 for (HloComputation* computation : module->computations()) { in Run()
Dcompile_only_service.h53 const absl::Span<const AotXlaComputationInstance> computations,
58 const absl::Span<const AotXlaComputationInstance> computations,
Ddynamic_index_splitter.cc35 std::vector<HloComputation*> computations = in Run() local
37 for (HloComputation* computation : computations) { in Run()
Dcompile_only_service.cc66 const absl::Span<const AotXlaComputationInstance> computations, in CompileAheadOfTime() argument
70 for (const AotXlaComputationInstance& instance : computations) { in CompileAheadOfTime()
Dhlo_domain_verifier.cc49 for (HloComputation* computation : module_->computations()) { in PopulateDomainKinds()
68 for (HloComputation* computation : module_->computations()) { in Run()
Dhlo_module.h152 computations() const { in computations() function
158 computations() { in computations() function
Dcopy_insertion.cc565 for (HloComputation* computation : module.computations()) { in CreateCopyMap()
969 for (HloComputation* computation : module->computations()) { in AddCopiesToResolveInterference()
1025 for (HloComputation* computation : module->computations()) { in AddSpecialCaseCopies()
1114 for (HloComputation* computation : module->computations()) { in RemoveUnnecessaryCopies()
1160 for (HloComputation* computation : module->computations()) { in Run()
1201 for (HloComputation* computation : module->computations()) { in Run()
1250 for (HloComputation* computation : module->computations()) { in AddCopiesForBufferAssignment()
/external/tensorflow/tensorflow/compiler/xla/client/
Dcompile_only_client.cc26 const absl::Span<const AotXlaComputationInstance> computations, in CompileAheadOfTime() argument
30 service_instances.reserve(computations.size()); in CompileAheadOfTime()
31 for (const AotXlaComputationInstance& instance : computations) { in CompileAheadOfTime()
Dclient.cc329 absl::Span<const XlaComputationInstance> computations) { in ExecuteParallel() argument
332 for (const XlaComputationInstance& computation : computations) { in ExecuteParallel()
356 if (i < computations.size() && in ExecuteParallel()
357 computations[i].execution_profile != nullptr) { in ExecuteParallel()
358 *computations[i].execution_profile = response.responses(i).profile(); in ExecuteParallel()
/external/tensorflow/tensorflow/compiler/xla/rpc/
Dxla_service.proto18 // Users 1) build up computations and 2) create allocations via this API.
22 // Users build up computations using a ComputationHandle, and talk about
38 // All data in XLA computations are conceptually immutable.
118 // handles can be used to specify the device on which to execute computations
125 // two computations via a pair of Send and Recv instructions.
139 // Invokes the provided list of computations in parallel with the provided
/external/tensorflow/tensorflow/compiler/xla/service/g3doc/
Dhlo_parser.md5 : 'HloModule' name computations
10 computations
12 | computation computations
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_sanitize_constant_names.cc37 for (HloComputation* computation : module->computations()) { in Run()
54 for (HloComputation* computation : module->computations()) { in Run()
/external/tensorflow/tensorflow/compiler/xla/tools/
Dinteractive_graphviz.cc117 for (const auto& computation : module.computations()) { in FindInstruction()
133 for (auto* computation : module.computations()) { in FindComputation()
193 std::cout << R"(Illegal syntax; "list computations" takes no arguments.)"; in DoListComputationsCommand()
203 for (const auto& computation : module.computations()) { in DoListComputationsCommand()
227 for (const auto& computation : module.computations()) { in DoListCommand()
287 for (const HloComputation* c : module.computations()) { in DoInfoCommand()
/external/swiftshader/third_party/subzero/tests_lit/llvm2ice_tests/
Dcmp-opt.ll34 ; Two bool computations
46 ; Two bool computations
/external/llvm/test/CodeGen/Mips/cstmaterialization/
Dstack.ll32 ; These are here to match other lui's used in address computations. We need to
33 ; investigate why address computations are not CSE'd. Or implement it.
/external/swiftshader/third_party/llvm-7.0/llvm/test/CodeGen/Mips/cstmaterialization/
Dstack.ll32 ; These are here to match other lui's used in address computations. We need to
33 ; investigate why address computations are not CSE'd. Or implement it.
/external/tensorflow/tensorflow/compiler/xla/tests/
Dhlo_test_base.cc499 auto computations = module->computations(); in FindComputation() local
501 computations, [&](HloComputation* c) { return c->name() == name; }); in FindComputation()
502 if (it == computations.end()) { in FindComputation()
510 for (const HloComputation* c : module->computations()) { in FindInstruction()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_BroadcastGradientArgs.pbtxt6 This is typically used by gradient computations for a broadcasting operation.
/external/tensorflow/tensorflow/compiler/xla/
DREADME.md6 algebra that optimizes TensorFlow computations. See the [documentation](./g3doc/overview.md).

123456789