Searched refs:UseKind (Results 1 – 2 of 2) sorted by relevance
2865 using UseKind = HloInstruction::UseKind; typedef in xla::HloInstruction::FusionReusesParamElements2871 static UseKind Compute(int64 i, const HloInstruction& hlo) { in Compute()2872 absl::flat_hash_map<const HloInstruction*, UseKind> memoization_cache; in Compute()2877 static UseKind ComputeInternal( in ComputeInternal()2879 absl::flat_hash_map<const HloInstruction*, UseKind>* cache) { in ComputeInternal()2882 return UseKind::kUse; in ComputeInternal()2886 auto p = cache->emplace(&hlo, UseKind{}); in ComputeInternal()2892 UseKind old_val = value_it->second; in ComputeInternal()2895 UseKind new_val = in ComputeInternal()2910 static UseKind Plus(UseKind a, UseKind b) { in Plus()[all …]
1251 return OperandElementUse(i) == UseKind::kReuse; in ReusesOperandElements()1621 enum class UseKind { kNoUse, kReuse, kUsePermutingElements, kUse }; enum1728 UseKind OperandElementUse(int64 i) const;