Searched refs:while_nest_multiplier (Results 1 – 2 of 2) sorted by relevance
158 float while_nest_multiplier; in GetAlternateMemoryBenefit() local161 auto it = cache->while_nest_multiplier.find(&instruction); in GetAlternateMemoryBenefit()162 if (it != cache->while_nest_multiplier.end()) { in GetAlternateMemoryBenefit()163 while_nest_multiplier = it->second; in GetAlternateMemoryBenefit()165 while_nest_multiplier = tensorflow::MathUtil::IPow<float>( in GetAlternateMemoryBenefit()169 cache->while_nest_multiplier[&instruction] = while_nest_multiplier; in GetAlternateMemoryBenefit()172 while_nest_multiplier = tensorflow::MathUtil::IPow<float>( in GetAlternateMemoryBenefit()178 while_nest_multiplier; in GetAlternateMemoryBenefit()
85 absl::flat_hash_map<const HloInstruction*, float> while_nest_multiplier; member