Home
last modified time | relevance | path

Searched refs:kFlopsKey (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.cc30 constexpr char HloCostAnalysis::kFlopsKey[]; member in xla::HloCostAnalysis
102 current_properties_[kFlopsKey] = computation_count; in HandleElementwiseOp()
250 current_properties_[kFlopsKey] = in HandleDot()
531 current_properties_[kFlopsKey] = fma_count * kFmaFlops; in HandleConvolution()
545 current_properties_[kFlopsKey] = kFmaFlops * kFmaPerComplexMul * log_factors * in HandleFft()
560 current_properties_[kFlopsKey] = kFmaFlops * elems; in HandleTriangularSolve()
572 current_properties_[kFlopsKey] = elems / 3; in HandleCholesky()
589 current_properties_[kFlopsKey] = flops; in HandleAllReduce()
650 current_properties_[kFlopsKey] = -1; in HandleCustomCall()
659 current_properties_[kFlopsKey] = elements * tensorflow::Log2Ceiling(elements); in HandleSort()
[all …]
Dhlo_cost_analysis.h42 static constexpr char kFlopsKey[] = "flops"; variable
123 per_second_rates_[kFlopsKey] = value; in set_flops_per_second()