Searched refs:kFlopsKey (Results 1 – 2 of 2) sorted by relevance
30 constexpr char HloCostAnalysis::kFlopsKey[]; member in xla::HloCostAnalysis102 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 …]
42 static constexpr char kFlopsKey[] = "flops"; variable123 per_second_rates_[kFlopsKey] = value; in set_flops_per_second()