Searched refs:HloCostAnalysis (Results 1 – 19 of 19) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | 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) 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 …]
|
D | hlo_cost_analysis_test.cc | 147 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 …]
|
D | hlo_cost_analysis.h | 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);
|
D | compiler.h | 246 virtual HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const = 0; 251 HloCostAnalysis::ShapeSizeFunction shape_size = ShapeSizeBytesFunction(); in BufferSizeBytesFunction()
|
D | hlo_execution_profile_test.cc | 54 HloCostAnalysis cost_analysis(shape_size_function); in TEST_F()
|
D | hlo_execution_profile.h | 100 const HloCostAnalysis& cost_analysis, const string& entry_computation_name);
|
D | hlo_execution_profile.cc | 51 const HloCostAnalysis& cost_analysis, in CreateHloProfilePrinterData()
|
D | hlo_profile_printer_data.proto | 31 // Metrics computed by HloCostAnalysis.
|
D | service.cc | 1142 HloCostAnalysis analysis( in GetComputationGraphStats()
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | parallel_task_assignment.cc | 33 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()
|
D | parallel_task_assignment.h | 42 const HloCostAnalysis::ShapeSizeFunction& shape_size, 69 const HloCostAnalysis::ShapeSizeFunction& shape_size, in ParallelTaskAssigner() 102 HloCostAnalysis::ShapeSizeFunction shape_size_function_;
|
D | parallel_task_assignment_test.cc | 28 const HloCostAnalysis::ShapeSizeFunction shape_size_func_ =
|
D | cpu_compiler.h | 150 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override;
|
D | cpu_compiler.cc | 512 HloCostAnalysis cost_analysis(shape_size_bytes); in CreateHloProfilingArtifacts() 946 HloCostAnalysis::ShapeSizeFunction CpuCompiler::ShapeSizeBytesFunction() const { in ShapeSizeBytesFunction()
|
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
D | compiler.h | 71 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override;
|
D | compiler.cc | 184 HloCostAnalysis::ShapeSizeFunction InterpreterCompiler::ShapeSizeBytesFunction() in ShapeSizeBytesFunction()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | nvptx_compiler.h | 67 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override { in ShapeSizeBytesFunction()
|
D | nvptx_compiler.cc | 778 HloCostAnalysis cost_analysis(ShapeSizeBytesFunction()); in RunBackend()
|
/external/tensorflow/ |
D | RELEASE.md | 1070 * Add time estimation to HloCostAnalysis.
|