Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.cc335 kFmaFlops * ShapeUtil::ElementsIn(dot_shape) * reduction_width; in HandleDot()
690 current_properties_[kFlopsKey] = fma_count * kFmaFlops; in HandleConvolution()
704 current_properties_[kFlopsKey] = kFmaFlops * kFmaPerComplexMul * log_factors * in HandleFft()
724 current_properties_[kFlopsKey] = kFmaFlops * elems; in HandleTriangularSolve()
Dhlo_cost_analysis.h196 static constexpr int64 kFmaFlops = 2; variable