Searched refs:OperandIndices (Results 1 – 14 of 14) sorted by relevance
28 using OperandIndices = std::vector<int64>;32 using FoldableOperands = std::function<OperandIndices(const HloInstruction&,33 const OperandIndices&)>;38 static OperandIndices NeverFoldTranspose(const HloInstruction&, in NeverFoldTranspose()39 const OperandIndices&) { in NeverFoldTranspose() argument
34 TransposeFolding::OperandIndices CanFoldOperandsIntoDot( in CanFoldOperandsIntoDot()43 TransposeFolding::OperandIndices operand_set; in CanFoldOperandsIntoDot()56 TransposeFolding::OperandIndices CanFoldOperandsIntoConvolution( in CanFoldOperandsIntoConvolution()64 TransposeFolding::OperandIndices operand_set; in CanFoldOperandsIntoConvolution()76 std::pair<HloInstruction*, TransposeFolding::OperandIndices>;198 std::vector<std::pair<HloInstruction*, OperandIndices>> foldable_dots; in Run()199 std::vector<std::pair<HloInstruction*, OperandIndices>> foldable_convolutions; in Run()203 OperandIndices operand_indices = in Run()210 OperandIndices operand_indices = CanFoldOperandsIntoConvolution( in Run()
49 const TransposeFolding::OperandIndices& candidate_operands) { in FoldTranspose()53 const TransposeFolding::OperandIndices& candidate_operands) { in FoldTranspose()98 const TransposeFolding::OperandIndices& candidate_operands) { in TEST_F()102 const TransposeFolding::OperandIndices& candidate_operands) { in TEST_F()126 const TransposeFolding::OperandIndices& candidate_operands) { in TEST_F()130 const TransposeFolding::OperandIndices& candidate_operands) { in TEST_F()
647 for (int64 op_idx : alias_user->OperandIndices(alias.instruction())) { in GetAllUsesOfInstructionAtIndex()666 if (fusion->OperandIndices(operand).size() > 1) { in HasUniqueFusedUseOfOperandAt()766 std::vector<int64> operand_indices = user->OperandIndices(operand); in CanShareOperandBufferWithUser()781 std::vector<int64> operand_indices = user->OperandIndices(operand); in CanShareOperandBufferWithUser()789 std::vector<int64> operand_indices = user->OperandIndices(operand); in CanShareOperandBufferWithUser()796 const std::vector<int64> operand_indices = user->OperandIndices(operand); in CanShareOperandBufferWithUser()
64 for (const int64 use_index : user->OperandIndices(current)) { in AreTransitiveUsesElementwiseOrTuple()693 for (int64 operand_number : user->OperandIndices(instruction)) { in Propagate()1099 std::vector<int64> operand_indices = user->OperandIndices(operand); in CanShareOperandBufferWithUser()1114 std::vector<int64> operand_indices = user->OperandIndices(operand); in CanShareOperandBufferWithUser()
180 for (int64 operand_number : user->OperandIndices(position.instruction)) { in SetPositionsAndComputeUses()
1257 std::vector<int64> OperandIndices(const HloInstruction* operand) const;
1998 DCHECK(!instruction->OperandIndices(operand).empty()); in OutputIsPermutationOfOperandElements()2015 std::vector<int64> operand_indices = instruction->OperandIndices(operand); in OutputIsSubsetOfOperandElements()
1184 user->OperandIndices(buffer_alias.instruction())) { in GetArrayUsesOfBuffer()
45 std::vector<int64> operand_indices = instruction->OperandIndices(operand); in IsInstructionElementwiseOnOperand()
2838 std::vector<int64> HloInstruction::OperandIndices( in OperandIndices() function in xla::HloInstruction
246 const TransposeFolding::OperandIndices& candidate_operands) { in OptimizeHloModule()248 : TransposeFolding::OperandIndices{}; in OptimizeHloModule()
186 const TransposeFolding::OperandIndices& candidate_operands) { in TEST_F()216 const TransposeFolding::OperandIndices& candidate_operands) { in TEST_F()247 const TransposeFolding::OperandIndices& candidate_operands) { in TEST_F()
314 const TransposeFolding::OperandIndices& candidate_operands) { in RunHloPassesThroughLayoutAssn()318 : TransposeFolding::OperandIndices{}; in RunHloPassesThroughLayoutAssn()