HomeSort by relevance Sort by last modified time
    Searched refs:GetValueSet (Results 1 - 7 of 7) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_dataflow_analysis.h 105 const HloValueSet& GetValueSet(const HloInstruction* instruction,
107 const HloValueSet& GetValueSet(const HloPosition& position) const;
108 HloValueSet& GetValueSet(const HloPosition& position);
109 HloValueSet& GetValueSet(const HloInstruction* instruction,
116 return GetValueSet(instruction, index).GetUniqueValue();
120 return GetValue(GetValueSet(instruction, index).GetUniqueValue().id());
hlo_dataflow_analysis.cc 80 const HloValueSet& value_set = GetValueSet(instruction, index);
167 GetValueSet(instruction, /*index=*/{});
287 const HloValueSet& HloDataflowAnalysis::GetValueSet(
292 HloValueSet& HloDataflowAnalysis::GetValueSet(const HloInstruction* instruction,
297 const HloValueSet& HloDataflowAnalysis::GetValueSet(
299 return GetValueSet(position.instruction, position.index);
302 HloValueSet& HloDataflowAnalysis::GetValueSet(const HloPosition& position) {
303 return GetValueSet(position.instruction, position.index);
331 HloValueSet& value_set = GetValueSet(send, index);
353 GetValueSet(recv_done->operand(0), index)
    [all...]
bfloat16_propagation.cc 91 dataflow_->GetValueSet(root, entry.second).values()) {
112 dataflow_->GetValueSet(inst, index).values()) {
214 auto& value_set = dataflow_->GetValueSet(&hlo, index);
346 for (const auto* value : dataflow_->GetValueSet(hlo, index).values()) {
472 for (const auto* value : dataflow_->GetValueSet(root, index).values()) {
518 for (const auto* value : dataflow_->GetValueSet(hlo, index).values()) {
534 for (const auto* value : dataflow_->GetValueSet(hlo, index).values()) {
    [all...]
copy_insertion.cc 179 if (dataflow.GetValueSet(init, index).values().size() > 1 ||
180 dataflow.GetValueSet(xla_while, index).values().size() > 1) {
571 dataflow_.GetValueSet(instruction->operand(0));
    [all...]
hlo_alias_analysis.cc 364 dataflow_analysis_->GetValueSet(instruction, index).values()) {
hlo_alias_analysis_test.cc 62 .GetValueSet(instruction, index)
    [all...]
hlo_dataflow_analysis_test.cc 65 analysis_->GetValueSet(instruction, index).values()) {
    [all...]

Completed in 73 milliseconds