Searched refs:kFlopsKey (Results 1 – 3 of 3) sorted by relevance
35 constexpr const char HloCostAnalysis::kFlopsKey[]; member in xla::HloCostAnalysis115 current_properties_[kFlopsKey] = computation_count; in HandleElementwiseOp()334 current_properties_[kFlopsKey] = 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()740 current_properties_[kFlopsKey] = elems / 3; in HandleCholesky()761 current_properties_[kFlopsKey] = flops; in HandleAllReduce()947 current_properties_[kFlopsKey] = -1; in HandleCustomCall()956 current_properties_[kFlopsKey] = elements * tensorflow::Log2Ceiling(elements); in HandleSort()[all …]
43 static constexpr const char kFlopsKey[] = "flops"; variable138 per_second_rates_[kFlopsKey] = value; in set_flops_per_second()
252 cost_analysis_.per_second_rate(HloCostAnalysis::kFlopsKey), in GetInstructionElapsedDueToCompute()