Home
last modified time | relevance | path

Searched refs:new_computation (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dflatten_call_graph.cc35 HloComputation* new_computation) { in ReplaceCalledComputation() argument
39 instruction->set_while_condition(new_computation); in ReplaceCalledComputation()
42 instruction->set_while_body(new_computation); in ReplaceCalledComputation()
48 instruction->set_to_apply(new_computation); in ReplaceCalledComputation()
57 instruction->set_branch_computation(b, new_computation); in ReplaceCalledComputation()
Drng_bit_generator_expander.cc95 HloComputation* new_computation = in GetGeneratorComputation() local
97 computation_cache_.emplace(cache_key, new_computation); in GetGeneratorComputation()
98 return new_computation; in GetGeneratorComputation()
Dhlo_clone_context.h50 HloComputation* new_computation) { in MapComputation() argument
51 computations_[old_computation] = new_computation; in MapComputation()
Dhlo_module.cc197 HloComputation* new_computation = tensorflow::gtl::FindWithDefault( in ReplaceComputations() local
199 if (new_computation != nullptr) { in ReplaceComputations()
200 instruction->set_branch_computation(b, new_computation); in ReplaceComputations()
760 HloComputation* new_computation; in DeepCloneComputation() local
762 if ((new_computation = context->FindComputation(computation)) != nullptr) { in DeepCloneComputation()
763 return new_computation; in DeepCloneComputation()
765 new_computation = in DeepCloneComputation()
768 new_computation = AddEmbeddedComputation(computation->Clone("")); in DeepCloneComputation()
770 return new_computation; in DeepCloneComputation()
Dconditional_simplifier.cc241 HloComputation* new_computation = in TryRemoveUnusedConditionalOperands() local
243 param = new_computation->parameter_instruction(0); in TryRemoveUnusedConditionalOperands()
262 conditional->set_branch_computation(branch, new_computation); in TryRemoveUnusedConditionalOperands()
Ddynamic_dimension_inference.cc1308 HloComputation* new_computation = branch_computation; in HandleConditional() local
1315 new_computation, in HandleConditional()
1338 new_computation, dynamic_parameter_binding, parent_)); in HandleConditional()
1344 new_computation->root_instruction(), in HandleConditional()
1360 new_computation->root_instruction(), hlos_to_add_in_root); in HandleConditional()
1361 new_computation->set_root_instruction(new_branch_root, in HandleConditional()
1365 new_branch_computations.push_back(new_computation); in HandleConditional()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc4014 HloComputationProto new_computation(e); in AddCalledComputation() local
4016 remapped_ids[new_computation.id()] = computation_id; in AddCalledComputation()
4017 SetProtoIdAndName(&new_computation, in AddCalledComputation()
4018 GetBaseName(new_computation.name(), kNameSeparator), in AddCalledComputation()
4020 for (auto& instruction : *new_computation.mutable_instructions()) { in AddCalledComputation()
4027 new_computation.set_root_id(remapped_ids.at(new_computation.root_id())); in AddCalledComputation()
4029 imported_computations.push_back(std::move(new_computation)); in AddCalledComputation()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc3392 auto new_computation = in DoPartition() local
3395 DoCodeMotionForWindowedDotGeneralLoops(new_computation, options)); in DoPartition()
3399 replacement[computation] = new_computation; in DoPartition()