Searched refs:kBytesAccessedKey (Results 1 – 3 of 3) sorted by relevance
37 constexpr const char HloCostAnalysis::kBytesAccessedKey[]; member in xla::HloCostAnalysis64 current_properties_[kBytesAccessedKey] = bytes_accessed; in Preprocess()210 current_properties_[kBytesAccessedKey] = 0; in HandleParameter()218 current_properties_[kBytesAccessedKey] = 0; in HandleConstant()233 current_properties_[kBytesAccessedKey] = 0; in HandleGetTupleElement()253 current_properties_[kBytesAccessedKey] = GetShapeSize(slice->shape()) * 2; in HandleSlice()261 current_properties_[kBytesAccessedKey] = in HandleDynamicSlice()272 current_properties_[kBytesAccessedKey] = in HandleDynamicUpdateSlice()291 current_properties_[kBytesAccessedKey] = GetShapeSize(tuple->shape()); in HandleTuple()314 current_properties_[kBytesAccessedKey] = 0; in HandleDomain()[all …]
45 static constexpr const char kBytesAccessedKey[] = "bytes accessed"; variable144 per_second_rates_[kBytesAccessedKey] = value; in set_bytes_per_second()
264 cost_analysis_.per_second_rate(HloCostAnalysis::kBytesAccessedKey); in GetInstructionElapsedDueToMemory()276 cost_analysis_.per_second_rate(HloCostAnalysis::kBytesAccessedKey); in GetInstructionElapsedDueToMemory()289 cost_analysis_.per_second_rate(HloCostAnalysis::kBytesAccessedKey); in GetInstructionElapsedDueToMemory()