Searched defs:cost_analysis (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | parallel_task_assignment.cc | 58 std::unique_ptr<HloCostAnalysis> cost_analysis) in DefaultCostModel()
119 auto cost_analysis = std::make_unique<HloCostAnalysis>(shape_size); in ParallelTaskAssignment() local
|
| H A D | cpu_compiler.cc | 807 HloCostAnalysis cost_analysis(shape_size_bytes); in CreateHloProfilingArtifacts() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/analysis/ |
| H A D | test_cost_analysis_pass.cc | 32 const auto& cost_analysis = getAnalysis<CostAnalysis>(); in runOnOperation() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | memory_space_assignment.h | 127 const HloCostAnalysis& cost_analysis() const { return cost_analysis_; } in cost_analysis() function
211 const HloCostAnalysis& cost_analysis, const Options& options, in MemorySpaceAssignmentCostAnalysis()
1028 MemorySpaceAssignmentCostAnalysis* cost_analysis = nullptr; member
|
| H A D | hlo_execution_profile_test.cc | 53 HloCostAnalysis cost_analysis(shape_size_function); in TEST_F() local
|
| H A D | hlo_execution_profile.cc | 52 const HloCostAnalysis& cost_analysis, in CreateHloProfilePrinterData()
|
| H A D | memory_space_assignment.cc | 246 MemorySpaceAssignmentCostAnalysis::Create(const HloCostAnalysis& cost_analysis, in Create()
551 const MemorySpaceAssignmentCostAnalysis& cost_analysis, in CostAnalysisPrefetchIntervalPicker()
3432 const MemorySpaceAssignmentCostAnalysis& cost_analysis, in GetMemoryBoundednessBufferIntervalCompare()
|
| H A D | memory_space_assignment_test.cc | 76 auto cost_analysis = MemorySpaceAssignmentCostAnalysis::Create( in AssignMemorySpaceUsingCostAnalysis() local
353 Create(const HloCostAnalysis& cost_analysis, const HloModule& module, in Create()
416 const HloCostAnalysis& cost_analysis, const Options& options, in FakeMemorySpaceAssignmentCostAnalysis()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| H A D | tf_to_tfrt.cc | 118 const tfrt_compiler::CostAnalysis *cost_analysis, in FallbackExecuteOpConversion()
1531 const tfrt_compiler::CostAnalysis *cost_analysis, in PopulateTFToTFRTConversionPatterns()
1668 tfrt_compiler::CostAnalysis cost_analysis(func); in runOnFunction() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | gpu_compiler.cc | 1346 GpuHloCostAnalysis cost_analysis(options); in RunBackend() local
|