HomeSort by relevance Sort by last modified time
    Searched refs:kFlopsKey (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_cost_analysis.cc 30 constexpr char HloCostAnalysis::kFlopsKey[];
102 current_properties_[kFlopsKey] = computation_count;
250 current_properties_[kFlopsKey] =
531 current_properties_[kFlopsKey] = fma_count * kFmaFlops;
545 current_properties_[kFlopsKey] = kFmaFlops * kFmaPerComplexMul * log_factors *
560 current_properties_[kFlopsKey] = kFmaFlops * elems;
572 current_properties_[kFlopsKey] = elems / 3;
589 current_properties_[kFlopsKey] = flops;
650 current_properties_[kFlopsKey] = -1;
659 current_properties_[kFlopsKey] = elements * tensorflow::Log2Ceiling(elements)
    [all...]
hlo_cost_analysis.h 42 static constexpr char kFlopsKey[] = "flops";
123 per_second_rates_[kFlopsKey] = value;

Completed in 189 milliseconds