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
|