Searched refs:hlo_properties_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_cost_analysis.cc | 85 TF_RET_CHECK(hlo_properties_.emplace(hlo, current_properties_).second); in Postprocess()
1072 return GetPropertyForHlo(hlo, kFlopsKey, hlo_properties_); in flop_count()
1076 return GetPropertyForHlo(hlo, kTranscendentalsKey, hlo_properties_); in transcendental_count()
1080 return GetPropertyForHlo(hlo, kBytesAccessedKey, hlo_properties_); in bytes_accessed()
1087 hlo_properties_); in operand_bytes_accessed()
1093 hlo_properties_); in output_bytes_accessed()
1097 return GetPropertyForHlo(hlo, kOptimalSecondsKey, hlo_properties_); in optimal_seconds()
1141 hlo_properties_.insert(visitor->hlo_properties_.begin(), in ProcessSubcomputation()
1142 visitor->hlo_properties_.end()); in ProcessSubcomputation()
|
| D | hlo_cost_analysis.h | 243 HloToProperties hlo_properties_; variable
|