Searched refs:kFlopsKey (Results 1 – 2 of 2) sorted by relevance
29 constexpr char HloCostAnalysis::kFlopsKey[]; member in xla::HloCostAnalysis100 current_properties_[kFlopsKey] = computation_count; in HandleElementwiseOp()220 current_properties_[kFlopsKey] = kFmaFlops * fma_count; in HandleDot()394 current_properties_[kFlopsKey] = in HandleConvolution()409 current_properties_[kFlopsKey] = kFmaFlops * kFmaPerComplexMul * log_factors * in HandleFft()427 current_properties_[kFlopsKey] = flops; in HandleCrossReplicaSum()489 current_properties_[kFlopsKey] = elements * tensorflow::Log2Ceiling(elements); in HandleSort()546 return GetProperty(kFlopsKey, properties_sum_); in flop_count()562 return GetPropertyForHlo(hlo, kFlopsKey, hlo_properties_); in flop_count()
42 static constexpr char kFlopsKey[] = "flops"; variable112 per_second_rates_[kFlopsKey] = value; in set_flops_per_second()