Home
last modified time | relevance | path

Searched refs:GetInstructionValueSet (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_dataflow_analysis.cc200 GetInstructionValueSet(instruction) in ToString()
234 << GetInstructionValueSet(instruction).ToString(); in Phi()
255 for (auto& pair : GetInstructionValueSet(instruction)) { in Phi()
356 GetInstructionValueSet(instruction); in GetFlattenedValueSet()
370 return GetInstructionValueSet(instruction).element(index); in GetValueSet()
375 return *GetInstructionValueSet(instruction).mutable_element(index); in GetValueSet()
390 GetInstructionValueSet(bitcast->operand(0)); in UpdateBitcastValueSet()
391 InstructionValueSet& bitcast_set = GetInstructionValueSet(bitcast); in UpdateBitcastValueSet()
403 GetInstructionValueSet(set_dimension_size->operand(0)); in UpdateSetDimensionSizeValueSet()
405 GetInstructionValueSet(set_dimension_size); in UpdateSetDimensionSizeValueSet()
[all …]
Dhlo_dataflow_analysis.h102 const InstructionValueSet& GetInstructionValueSet(
104 InstructionValueSet& GetInstructionValueSet(
Dhlo_alias_analysis.cc413 dataflow_analysis_->GetInstructionValueSet(instruction)) { in InstructionBuffersAreAmbiguous()
431 dataflow_analysis_->GetInstructionValueSet(instruction)) { in InstructionBuffersAreDistinct()
Dcopy_insertion.cc1130 alias_analysis->dataflow_analysis().GetInstructionValueSet(root)) { in AddSpecialCaseCopies()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.cc409 return assignment_->dataflow_analysis().GetInstructionValueSet( in GetRootValueSet()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.cc589 assignment.dataflow_analysis().GetInstructionValueSet(root); in GetOutputInfo()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtest_utils.cc504 for (const auto& pair : dataflow.GetInstructionValueSet(&param)) { in FindConstrainedUses()