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

  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_cost_analysis.h 32 // HloCostAnalysis traverses an HLO graph and calculates the amount of
37 class HloCostAnalysis : public ConstDfsHloVisitor {
50 explicit HloCostAnalysis(const ShapeSizeFunction& shape_size);
145 HloCostAnalysis(const ShapeSizeFunction& shape_size,
194 TF_DISALLOW_COPY_AND_ASSIGN(HloCostAnalysis);
hlo_cost_analysis.cc 29 constexpr char HloCostAnalysis::kFlopsKey[];
30 constexpr char HloCostAnalysis::kTranscendentalsKey[];
31 constexpr char HloCostAnalysis::kBytesAccessedKey[];
32 constexpr char HloCostAnalysis::kOptimalSecondsKey[];
34 HloCostAnalysis::HloCostAnalysis(const ShapeSizeFunction& shape_size)
35 : HloCostAnalysis(shape_size, {}) {}
37 HloCostAnalysis::HloCostAnalysis(const ShapeSizeFunction& shape_size,
41 Status HloCostAnalysis::Preprocess(const HloInstruction* hlo)
    [all...]

Completed in 272 milliseconds