| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_subcomputation_unification.cc | 26 const auto& computations = module->computations(); local
27 for (auto i = computations.begin(); i != computations.end(); ++i) {
28 for (auto j = computations.begin(); j != i; ++j) {
|
| while_loop_trip_count_annotator.cc | 24 for (const HloComputation* comp : module->computations()) {
|
| hlo_module_group_util.h | 57 // Returns the root instructions of the computations.
59 absl::Span<HloComputation* const> computations);
79 // different computations connected via communication instructions. During the
95 // Verifies that the computations are well-formed (e.g., no cycles).
96 Status VerifyComputations(absl::Span<HloComputation* const> computations);
99 // they can handle instructions across multiple computations.
101 // Creates the reachability map for the instructions in the computations.
103 absl::Span<HloComputation* const> computations);
|
| despecializer.cc | 34 for (HloComputation* computation : module->computations()) {
|
| compile_only_service.h | 48 // Compiles a list of xla computations for ahead-of-time execution. This is
53 const absl::Span<const AotXlaComputationInstance> computations,
58 const absl::Span<const AotXlaComputationInstance> computations,
|
| hlo_domain_isolator.cc | 31 for (HloComputation* computation : module->computations()) {
|
| compile_only_service.cc | 66 const absl::Span<const AotXlaComputationInstance> computations,
70 for (const AotXlaComputationInstance& instance : computations) {
|
| dynamic_index_splitter.cc | 35 std::vector<HloComputation*> computations = local
37 for (HloComputation* computation : computations) {
|
| hlo_module_group_util.cc | 196 absl::Span<HloComputation* const> computations) {
198 for (HloComputation* computation : computations) {
318 absl::Span<HloComputation* const> computations) {
326 for (HloComputation* computation : computations) {
349 absl::Span<HloComputation* const> computations) {
359 for (HloInstruction* root : RootInstructions(computations)) {
|
| hlo_domain_verifier.cc | 49 for (HloComputation* computation : module_->computations()) {
68 for (HloComputation* computation : module_->computations()) {
|
| conditional_simplifier.cc | 90 // we don't have to worry about mutating the lists of computations or
93 for (auto* comp : module->computations()) {
|
| hlo_get_dimension_size_rewriter.cc | 65 for (auto* computation : module->computations()) {
|
| hlo_module_dce.cc | 41 for (auto* computation : module->computations()) {
98 // computations to pass through tuple values (creating dead roots in while
|
| hlo_module_test.cc | 50 // Creates a computation which calls the given zero-parameter computations.
52 absl::Span<HloComputation* const> computations) {
54 for (auto computation : computations) {
74 // Create a module with two unconnected computations.
83 // We specified the same name for both computations, but the HloModule should
90 // Create and copy a module with a diamond call graph of computations.
156 // Create a module with a diamond call graph of computations.
305 // Serialize and deserialize and verify that the instruction and computations
315 // Verify that the computations and instructions all have the same unique id.
316 auto computation_copy_it = module_copy->computations().begin()
[all...] |
| hlo_module.cc | 85 // for computations and instructions created later. Also, set the
127 // Replace all uses of non-canonical computations with their
250 for (const HloComputation* computation : computations()) {
307 std::vector<std::unique_ptr<HloComputation>> computations; local
309 for (const HloComputationProto& computation_proto : proto.computations()) {
322 computations.push_back(std::move(computation));
328 // Sort the computations in the proto id's order.
329 absl::c_sort(computations, [&](const std::unique_ptr<HloComputation>& a,
334 // Add sorted computations to the module.
335 for (auto& computation : computations) {
[all...] |
| hlo_module.h | 57 // A module also contains some number of "nested computations". Each nested
82 // Replaces all uses of computations that are keys of 'replacements' with
87 // computations to replace. We could speed it up by keeping track of users of
88 // computations.
95 // Returns a deep copy of this module including all computations.
101 // the called computations as well. If the clone context is specified, it
143 // Gets the computations in this module.
148 // for (HloComputation* c : module->computations()) { ... }
152 computations() const { function in class:xla::HloModule
158 computations() { function in class:xla::HloModule
[all...] |
| hlo_domain_remover.cc | 63 for (HloComputation* computation : module_->computations()) {
|
| tuple_simplifier.cc | 39 for (auto* computation : module->computations()) {
|
| hlo_pass_pipeline_test.cc | 49 for (HloComputation* computation : module->computations()) {
68 for (HloComputation* computation : module->computations()) {
87 for (HloComputation* computation : module->computations()) {
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| gpu_sanitize_constant_names.cc | 37 for (HloComputation* computation : module->computations()) {
54 for (HloComputation* computation : module->computations()) {
|
| gpu_hlo_support_checker.cc | 25 for (auto* computation : module->computations()) {
|
| gpu_copy_insertion.cc | 46 for (HloComputation* computation : module->computations()) {
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_hlo_support_checker.cc | 25 for (auto* computation : module->computations()) {
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| compile_only_client.cc | 26 const absl::Span<const AotXlaComputationInstance> computations,
30 service_instances.reserve(computations.size());
31 for (const AotXlaComputationInstance& instance : computations) {
|
| compile_only_client.h | 49 // Compiles a list of xla computations for ahead-of-time execution.
55 const absl::Span<const AotXlaComputationInstance> computations,
|
