Home
last modified time | relevance | path

Searched refs:ResetComputationLayout (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dlayout_assignment.cc714 body_constraint->ResetComputationLayout( in AddMandatoryConstraints()
726 condition_constraint->ResetComputationLayout( in AddMandatoryConstraints()
1092 ->ResetComputationLayout(computed_computation_layout, in CopyOperandIfLayoutsDiffer()
2130 callee_constraint->ResetComputationLayout(callee_layout, priority, true, in CalculateComputationLayout()
H A Dlayout_assignment.h152 void ResetComputationLayout(const ComputationLayout& layout, int64_t priority, in ResetComputationLayout() function