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.cc251 kFmaFlops * ShapeUtil::ElementsIn(dot_shape) * reduction_width; 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()
Dhlo_cost_analysis.h154 static constexpr int64 kFmaFlops = 2; variable