Searched refs:ComputeBuffersAt (Results 1 – 4 of 4) sorted by relevance
383 std::vector<const HloBuffer*> buffers = ComputeBuffersAt(instruction, index); in GetUniqueBufferAt()395 std::vector<const HloBuffer*> HloAliasAnalysis::ComputeBuffersAt( in ComputeBuffersAt() function in xla::HloAliasAnalysis499 ComputeBuffersAt(instruction, index)) { in ToString()505 ComputeBuffersAt(instruction, /*index=*/{})) { in ToString()565 alias_analysis->ComputeBuffersAt(root, index)) { in Run()
75 std::vector<const HloBuffer*> ComputeBuffersAt(
196 for (const HloBuffer* buffer : alias_analysis_->ComputeBuffersAt( in GetMemoryBoundedness()1740 alias_analysis_.ComputeBuffersAt(parameter_instruction, index)) { in AddInputAndOutputRequiredAssignments()1764 alias_analysis_.ComputeBuffersAt(root_instruction, index)) { in AddInputAndOutputRequiredAssignments()2975 for (auto& buffer : alias_analysis->ComputeBuffersAt( in ExportAndColorBuffers()
1099 alias_analysis->ComputeBuffersAt(root, index); in AddSpecialCaseCopies()