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

  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_cost_analysis.cc 30 constexpr char HloCostAnalysis::kFlopsKey[];
31 constexpr char HloCostAnalysis::kTranscendentalsKey[];
32 constexpr char HloCostAnalysis::kBytesAccessedKey[];
33 constexpr char HloCostAnalysis::kOptimalSecondsKey[];
35 HloCostAnalysis::HloCostAnalysis(const ShapeSizeFunction& shape_size)
36 : HloCostAnalysis(shape_size, {}) {}
38 HloCostAnalysis::HloCostAnalysis(const ShapeSizeFunction& shape_size,
42 Status HloCostAnalysis::Preprocess(const HloInstruction* hlo)
    [all...]
hlo_cost_analysis_test.cc 48 // using the client computation builder and running the HloCostAnalysis that
147 HloCostAnalysis analysis(ShapeSize);
176 HloCostAnalysis analysis(ShapeSize);
205 HloCostAnalysis analysis(ShapeSize);
228 HloCostAnalysis analysis(ShapeSize);
249 HloCostAnalysis analysis(ShapeSize);
275 HloCostAnalysis analysis(ShapeSize);
304 HloCostAnalysis analysis(ShapeSize);
325 HloCostAnalysis analysis(ShapeSize);
343 HloCostAnalysis analysis(ShapeSize)
    [all...]
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);
156 HloCostAnalysis(const ShapeSizeFunction& shape_size,
223 TF_DISALLOW_COPY_AND_ASSIGN(HloCostAnalysis);
hlo_execution_profile_test.cc 54 HloCostAnalysis cost_analysis(shape_size_function);
compiler.h 246 virtual HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const = 0;
251 HloCostAnalysis::ShapeSizeFunction shape_size = ShapeSizeBytesFunction();
hlo_execution_profile.cc 51 const HloCostAnalysis& cost_analysis,
hlo_execution_profile.h 100 const HloCostAnalysis& cost_analysis, const string& entry_computation_name);
service.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
parallel_task_assignment.h 38 // 'shape_size': shape size function used by HloCostAnalysis during parallel
42 const HloCostAnalysis::ShapeSizeFunction& shape_size,
66 // 'shape_size': shape size function used by HloCostAnalysis during parallel
69 const HloCostAnalysis::ShapeSizeFunction& shape_size,
102 HloCostAnalysis::ShapeSizeFunction shape_size_function_;
parallel_task_assignment.cc 33 const HloCostAnalysis::ShapeSizeFunction& shape_size)
48 const HloCostAnalysis::ShapeSizeFunction shape_size_;
54 const HloCostAnalysis::ShapeSizeFunction& shape_size,
55 std::unique_ptr<HloCostAnalysis> cost_analysis)
103 const HloCostAnalysis::ShapeSizeFunction shape_size_;
104 const std::unique_ptr<HloCostAnalysis> cost_analysis_;
109 const HloCostAnalysis::ShapeSizeFunction& shape_size, HloModule* module,
114 auto cost_analysis = absl::make_unique<HloCostAnalysis>(shape_size);
123 // Note that HloCostAnalysis can returns an error status (likely because
124 // HLOs like CustomCall are not yet implemented in the HloCostAnalysis)
    [all...]
parallel_task_assignment_test.cc 28 const HloCostAnalysis::ShapeSizeFunction shape_size_func_ =
cpu_compiler.h 150 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override;
cpu_compiler.cc 512 HloCostAnalysis cost_analysis(shape_size_bytes);
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/interpreter/
compiler.h 71 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override;
compiler.cc 184 HloCostAnalysis::ShapeSizeFunction InterpreterCompiler::ShapeSizeBytesFunction()
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
nvptx_compiler.h 67 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override {
nvptx_compiler.cc     [all...]

Completed in 780 milliseconds