Searched refs:consider_using_bfloat16_ (Results 1 – 2 of 2) sorted by relevance
86 absl::flat_hash_set<const HloInstruction*> consider_using_bfloat16_; variable
431 if (!ContainsKey(consider_using_bfloat16_, hlo)) { in DetermineInstructionPrecision()461 !ContainsKey(consider_using_bfloat16_, hlo->operand(i))) { in InstructionIsCandidateForBF16Output()836 consider_using_bfloat16_.clear(); in Run()884 consider_using_bfloat16_.insert(inst); in Run()