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