Home
last modified time | relevance | path

Searched refs:HloCostAnalysis (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.cc30 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) in HloCostAnalysis() function in xla::HloCostAnalysis
36 : HloCostAnalysis(shape_size, {}) {} in HloCostAnalysis()
38 HloCostAnalysis::HloCostAnalysis(const ShapeSizeFunction& shape_size, in HloCostAnalysis() function in xla::HloCostAnalysis
42 Status HloCostAnalysis::Preprocess(const HloInstruction* hlo) { in Preprocess()
61 Status HloCostAnalysis::Postprocess(const HloInstruction* hlo) { in Postprocess()
85 Status HloCostAnalysis::HandleElementwiseOp( in HandleElementwiseOp()
[all …]
Dhlo_cost_analysis_test.cc147 HloCostAnalysis analysis(ShapeSize); in TEST_F()
176 HloCostAnalysis analysis(ShapeSize); in TEST_F()
205 HloCostAnalysis analysis(ShapeSize); in TEST_F()
228 HloCostAnalysis analysis(ShapeSize); in TEST_F()
249 HloCostAnalysis analysis(ShapeSize); in TEST_F()
275 HloCostAnalysis analysis(ShapeSize); in TEST_F()
304 HloCostAnalysis analysis(ShapeSize); in TEST_F()
325 HloCostAnalysis analysis(ShapeSize); in TEST_F()
343 HloCostAnalysis analysis(ShapeSize); in TEST_F()
362 HloCostAnalysis analysis(ShapeSize); in TEST_F()
[all …]
Dhlo_cost_analysis.h37 class HloCostAnalysis : public ConstDfsHloVisitor {
50 explicit HloCostAnalysis(const ShapeSizeFunction& shape_size);
156 HloCostAnalysis(const ShapeSizeFunction& shape_size,
223 TF_DISALLOW_COPY_AND_ASSIGN(HloCostAnalysis);
Dcompiler.h246 virtual HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const = 0;
251 HloCostAnalysis::ShapeSizeFunction shape_size = ShapeSizeBytesFunction(); in BufferSizeBytesFunction()
Dhlo_execution_profile_test.cc54 HloCostAnalysis cost_analysis(shape_size_function); in TEST_F()
Dhlo_execution_profile.h100 const HloCostAnalysis& cost_analysis, const string& entry_computation_name);
Dhlo_execution_profile.cc51 const HloCostAnalysis& cost_analysis, in CreateHloProfilePrinterData()
Dhlo_profile_printer_data.proto31 // Metrics computed by HloCostAnalysis.
Dservice.cc1142 HloCostAnalysis analysis( in GetComputationGraphStats()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dparallel_task_assignment.cc33 const HloCostAnalysis::ShapeSizeFunction& shape_size) in SimpleCostModel()
48 const HloCostAnalysis::ShapeSizeFunction shape_size_;
54 const HloCostAnalysis::ShapeSizeFunction& shape_size, in DefaultCostModel()
55 std::unique_ptr<HloCostAnalysis> cost_analysis) in DefaultCostModel()
103 const HloCostAnalysis::ShapeSizeFunction shape_size_;
104 const std::unique_ptr<HloCostAnalysis> cost_analysis_;
109 const HloCostAnalysis::ShapeSizeFunction& shape_size, HloModule* module, in ParallelTaskAssignment()
114 auto cost_analysis = absl::make_unique<HloCostAnalysis>(shape_size); in ParallelTaskAssignment()
Dparallel_task_assignment.h42 const HloCostAnalysis::ShapeSizeFunction& shape_size,
69 const HloCostAnalysis::ShapeSizeFunction& shape_size, in ParallelTaskAssigner()
102 HloCostAnalysis::ShapeSizeFunction shape_size_function_;
Dparallel_task_assignment_test.cc28 const HloCostAnalysis::ShapeSizeFunction shape_size_func_ =
Dcpu_compiler.h150 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override;
Dcpu_compiler.cc512 HloCostAnalysis cost_analysis(shape_size_bytes); in CreateHloProfilingArtifacts()
946 HloCostAnalysis::ShapeSizeFunction CpuCompiler::ShapeSizeBytesFunction() const { in ShapeSizeBytesFunction()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dcompiler.h71 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override;
Dcompiler.cc184 HloCostAnalysis::ShapeSizeFunction InterpreterCompiler::ShapeSizeBytesFunction() in ShapeSizeBytesFunction()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnvptx_compiler.h67 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override { in ShapeSizeBytesFunction()
Dnvptx_compiler.cc778 HloCostAnalysis cost_analysis(ShapeSizeBytesFunction()); in RunBackend()
/external/tensorflow/
DRELEASE.md1070 * Add time estimation to HloCostAnalysis.