Searched refs:hlo_properties_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_cost_analysis.cc | 77 TF_RET_CHECK(hlo_properties_.emplace(hlo, current_properties_).second); in Postprocess()
762 return GetPropertyForHlo(hlo, kFlopsKey, hlo_properties_); in flop_count()
766 return GetPropertyForHlo(hlo, kTranscendentalsKey, hlo_properties_); in transcendental_count()
770 return GetPropertyForHlo(hlo, kBytesAccessedKey, hlo_properties_); in bytes_accessed()
774 return GetPropertyForHlo(hlo, kOptimalSecondsKey, hlo_properties_); in optimal_seconds()
788 hlo_properties_.insert(visitor.hlo_properties_.begin(), in ProcessUnnestedSubcomputation()
789 visitor.hlo_properties_.end()); in ProcessUnnestedSubcomputation()
|
| D | hlo_cost_analysis.h | 204 HloToProperties hlo_properties_; variable
|