| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_cost_analysis.cc | 35 constexpr const char HloCostAnalysis::kFlopsKey[];
36 constexpr const char HloCostAnalysis::kTranscendentalsKey[];
37 constexpr const char HloCostAnalysis::kBytesAccessedKey[];
38 constexpr const char HloCostAnalysis::kOptimalSecondsKey[];
40 HloCostAnalysis::HloCostAnalysis(const ShapeSizeFunction& shape_size) in HloCostAnalysis() function in xla::HloCostAnalysis
41 : HloCostAnalysis(shape_size, {}) {} in HloCostAnalysis()
43 HloCostAnalysis::HloCostAnalysis(const ShapeSizeFunction& shape_size, in HloCostAnalysis() function in xla::HloCostAnalysis
47 Status HloCostAnalysis::Preprocess(const HloInstruction* hlo) { in Preprocess()
69 Status HloCostAnalysis::Postprocess(const HloInstruction* hlo) { in Postprocess()
93 Status HloCostAnalysis::HandleElementwiseOp( in HandleElementwiseOp()
[all …]
|
| D | hlo_cost_analysis_test.cc | 146 HloCostAnalysis analysis(ShapeSize); in TEST_F()
180 HloCostAnalysis analysis(ShapeSize); in TEST_F()
216 HloCostAnalysis analysis(ShapeSize); in TEST_F()
246 HloCostAnalysis analysis(ShapeSize); in TEST_F()
273 HloCostAnalysis analysis(ShapeSize); in TEST_F()
303 HloCostAnalysis analysis(ShapeSize); in TEST_F()
337 HloCostAnalysis analysis(ShapeSize); in TEST_F()
360 HloCostAnalysis analysis(ShapeSize); in TEST_F()
383 HloCostAnalysis analysis(ShapeSize); in TEST_F()
412 HloCostAnalysis analysis(ShapeSize); in TEST_F()
[all …]
|
| D | hlo_cost_analysis.h | 37 class HloCostAnalysis : public ConstDfsHloVisitor {
51 explicit HloCostAnalysis(const ShapeSizeFunction& shape_size);
198 HloCostAnalysis(const ShapeSizeFunction& shape_size,
201 virtual std::unique_ptr<HloCostAnalysis> CreateNestedCostAnalysis(
262 TF_DISALLOW_COPY_AND_ASSIGN(HloCostAnalysis);
|
| D | hlo_execution_profile_test.cc | 53 HloCostAnalysis cost_analysis(shape_size_function); in TEST_F()
|
| D | compiler.h | 291 virtual HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const = 0;
296 HloCostAnalysis::ShapeSizeFunction shape_size = ShapeSizeBytesFunction(); in BufferSizeBytesFunction()
|
| D | hlo_execution_profile.h | 100 const HloCostAnalysis& cost_analysis, const string& entry_computation_name);
|
| D | hlo_profile_printer_data.proto | 31 // Metrics computed by HloCostAnalysis.
|
| D | hlo_execution_profile.cc | 52 const HloCostAnalysis& cost_analysis, in CreateHloProfilePrinterData()
|
| D | memory_space_assignment.h | 91 const HloCostAnalysis& cost_analysis,
95 const HloCostAnalysis& cost_analysis() const { return cost_analysis_; } in cost_analysis()
156 const HloCostAnalysis& cost_analysis, in MemorySpaceAssignmentCostAnalysis()
172 const HloCostAnalysis& cost_analysis_;
|
| D | memory_space_assignment.cc | 135 const HloCostAnalysis& cost_analysis, in Create()
252 cost_analysis_.per_second_rate(HloCostAnalysis::kFlopsKey), in GetInstructionElapsedDueToCompute()
254 cost_analysis_.per_second_rate(HloCostAnalysis::kTranscendentalsKey)); in GetInstructionElapsedDueToCompute()
264 cost_analysis_.per_second_rate(HloCostAnalysis::kBytesAccessedKey); in GetInstructionElapsedDueToMemory()
276 cost_analysis_.per_second_rate(HloCostAnalysis::kBytesAccessedKey); in GetInstructionElapsedDueToMemory()
289 cost_analysis_.per_second_rate(HloCostAnalysis::kBytesAccessedKey); in GetInstructionElapsedDueToMemory()
|
| D | memory_space_assignment_test.cc | 48 HloCostAnalysis hlo_cost_analysis(ShapeSize); in AssignMemorySpaceUsingCostAnalysis()
321 Create(const HloCostAnalysis& cost_analysis, const HloModule& module) { in Create()
382 const HloCostAnalysis& cost_analysis, in FakeMemorySpaceAssignmentCostAnalysis()
4110 HloCostAnalysis hlo_cost_analysis(ShapeSize); in TEST_P()
5258 HloCostAnalysis hlo_cost_analysis(ShapeSize); in TEST_F()
5355 HloCostAnalysis hlo_cost_analysis(ShapeSize); in TEST_F()
5435 HloCostAnalysis hlo_cost_analysis(ShapeSize); in TEST_F()
5501 HloCostAnalysis hlo_cost_analysis(ShapeSize); in TEST_F()
|
| D | service.cc | 1119 HloCostAnalysis analysis( in GetComputationGraphStats()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | parallel_task_assignment.cc | 34 const HloCostAnalysis::ShapeSizeFunction& shape_size) in SimpleCostModel()
49 const HloCostAnalysis::ShapeSizeFunction shape_size_;
55 const HloCostAnalysis::ShapeSizeFunction& shape_size, in DefaultCostModel()
56 std::unique_ptr<HloCostAnalysis> cost_analysis) in DefaultCostModel()
105 const HloCostAnalysis::ShapeSizeFunction shape_size_;
106 const std::unique_ptr<HloCostAnalysis> cost_analysis_;
111 const HloCostAnalysis::ShapeSizeFunction& shape_size, HloModule* module, in ParallelTaskAssignment()
116 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 | 153 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override;
|
| D | cpu_compiler.cc | 562 HloCostAnalysis cost_analysis(shape_size_bytes); in CreateHloProfilingArtifacts()
1032 HloCostAnalysis::ShapeSizeFunction CpuCompiler::ShapeSizeBytesFunction() const { in ShapeSizeBytesFunction()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | compiler.h | 61 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override;
|
| D | compiler.cc | 165 HloCostAnalysis::ShapeSizeFunction InterpreterCompiler::ShapeSizeBytesFunction() in ShapeSizeBytesFunction()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_compiler.h | 112 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override { in ShapeSizeBytesFunction()
|
| D | gpu_compiler.cc | 884 HloCostAnalysis cost_analysis(ShapeSizeBytesFunction()); in RunBackend()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_client.h | 182 virtual StatusOr<std::unique_ptr<HloCostAnalysis>> GetHloCostAnalysis() = 0;
|
| D | pjrt_stream_executor_client.h | 175 StatusOr<std::unique_ptr<HloCostAnalysis>> GetHloCostAnalysis() override;
|
| D | pjrt_stream_executor_client.cc | 260 StatusOr<std::unique_ptr<HloCostAnalysis>>
262 return absl::make_unique<HloCostAnalysis>( in GetHloCostAnalysis()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_on_demand_compiler.cc | 358 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override { in ShapeSizeBytesFunction()
|