Searched refs:is_expensive (Results 1 – 5 of 5) sorted by relevance
42 std::function<bool(const HloInstruction& instruction)> is_expensive,46 : is_expensive_(is_expensive), in is_expensive_() argument124 bool is_expensive(const HloInstruction& instruction) { in is_expensive() function
54 inline int GetTFTraceMeLevel(bool is_expensive) { in GetTFTraceMeLevel() argument55 return is_expensive ? kInfo : kVerbose; in GetTFTraceMeLevel()
531 bool is_expensive) { in MightTrace() argument545 return profiler::TraceMe::Active(profiler::GetTFTraceMeLevel(is_expensive)); in MightTrace()558 const bool is_expensive = kernel_stats_->IsExpensive(item); in ProcessSync() local560 if (TF_PREDICT_FALSE(MightTrace(event_collector_, is_expensive))) { in ProcessSync()568 profiler::GetTFTraceMeLevel(is_expensive)); in ProcessSync()577 if (is_expensive || in ProcessSync()
68 if (producer->opcode() != HloOpcode::kFusion && is_expensive(*producer) && in ShouldFuseInexpensiveChecks()
97 if (producer->opcode() != HloOpcode::kFusion && is_expensive(*producer) && in ShouldFuse()