Home
last modified time | relevance | path

Searched defs:cost_analysis (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tfrt/analysis/
Dtest_cost_analysis_pass.cc25 const auto& cost_analysis = getAnalysis<CostAnalysis>(); in runOnOperation() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dparallel_task_assignment.cc56 std::unique_ptr<HloCostAnalysis> cost_analysis) in DefaultCostModel()
116 auto cost_analysis = absl::make_unique<HloCostAnalysis>(shape_size); in ParallelTaskAssignment() local
Dcpu_compiler.cc587 HloCostAnalysis cost_analysis(shape_size_bytes); in CreateHloProfilingArtifacts() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_execution_profile_test.cc53 HloCostAnalysis cost_analysis(shape_size_function); in TEST_F() local
Dhlo_execution_profile.cc52 const HloCostAnalysis& cost_analysis, in CreateHloProfilePrinterData()
Dmemory_space_assignment.h115 const HloCostAnalysis& cost_analysis() const { return cost_analysis_; } in cost_analysis() function
179 const HloCostAnalysis& cost_analysis, const Options& options, in MemorySpaceAssignmentCostAnalysis()
944 MemorySpaceAssignmentCostAnalysis* cost_analysis = nullptr; member
Dmemory_space_assignment.cc148 MemorySpaceAssignmentCostAnalysis::Create(const HloCostAnalysis& cost_analysis, in Create()
378 const MemorySpaceAssignmentCostAnalysis& cost_analysis, in CostAnalysisPrefetchIntervalPicker()
2863 const MemorySpaceAssignmentCostAnalysis& cost_analysis, in GetMemoryBoundednessBufferIntervalCompare()
Dmemory_space_assignment_test.cc74 auto cost_analysis = MemorySpaceAssignmentCostAnalysis::Create( in AssignMemorySpaceUsingCostAnalysis() local
339 Create(const HloCostAnalysis& cost_analysis, const HloModule& module, in Create()
402 const HloCostAnalysis& cost_analysis, const Options& options, in FakeMemorySpaceAssignmentCostAnalysis()
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
Dtf_to_tfrt.cc115 const tfrt_compiler::CostAnalysis *cost_analysis, in FallbackExecuteOpConversion()
1465 const tfrt_compiler::CostAnalysis *cost_analysis, in PopulateTFToTFRTConversionPatterns()
1601 tfrt_compiler::CostAnalysis cost_analysis(func); in runOnFunction() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_compiler.cc1060 HloCostAnalysis cost_analysis(ShapeSizeBytesFunction()); in RunBackend() local