Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_dataflow_analysis.cc369 GetInstructionValueSet(instruction) in ToString()
403 << GetInstructionValueSet(instruction).ToString(); in Phi()
424 for (auto& pair : GetInstructionValueSet(instruction)) { in Phi()
525 GetInstructionValueSet(instruction); in GetFlattenedValueSet()
539 return GetInstructionValueSet(instruction).element(index); in GetValueSet()
544 return *GetInstructionValueSet(instruction).mutable_element(index); in GetValueSet()
559 GetInstructionValueSet(bitcast->operand(0)); in UpdateBitcastValueSet()
560 InstructionValueSet& bitcast_set = GetInstructionValueSet(bitcast); in UpdateBitcastValueSet()
572 GetInstructionValueSet(set_dimension_size->operand(0)); in UpdateSetDimensionSizeValueSet()
574 GetInstructionValueSet(set_dimension_size); in UpdateSetDimensionSizeValueSet()
[all …]
Dhlo_dataflow_analysis.h116 const InstructionValueSet& GetInstructionValueSet(
118 InstructionValueSet& GetInstructionValueSet(
Dhlo_live_range.cc215 alias_analysis_.dataflow_analysis().GetInstructionValueSet( in CalculateBufferStartEndMap()
Dhlo_alias_analysis_test.cc108 analysis_->dataflow_analysis().GetInstructionValueSet(instruction)) { in InstructionBuffersAreAmbiguous()
127 analysis_->dataflow_analysis().GetInstructionValueSet(instruction)) { in InstructionBuffersAreDistinct()
Dcopy_insertion.cc1975 alias_analysis->dataflow_analysis().GetInstructionValueSet(root)) { in AddSpecialCaseCopies()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.cc400 return assignment_->dataflow_analysis().GetInstructionValueSet( in GetRootValueSet()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtest_utils.cc553 for (const auto& pair : dataflow.GetInstructionValueSet(&param)) { in FindConstrainedUses()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.cc971 assignment.dataflow_analysis().GetInstructionValueSet(root); in GetOutputInfo()
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtfrt_cpu_pjrt_client.cc285 return assignment.dataflow_analysis().GetInstructionValueSet( in GetRootValueSet()