Home
last modified time | relevance | path

Searched refs:cost_analysis (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_execution_profile.cc51 const HloCostAnalysis& cost_analysis, in CreateHloProfilePrinterData() argument
102 instruction_info->set_flop_count(cost_analysis.flop_count(*hlo)); in CreateHloProfilePrinterData()
104 cost_analysis.transcendental_count(*hlo)); in CreateHloProfilePrinterData()
105 instruction_info->set_bytes_accessed(cost_analysis.bytes_accessed(*hlo)); in CreateHloProfilePrinterData()
107 cost_analysis.optimal_seconds(*hlo)); in CreateHloProfilePrinterData()
Dhlo_execution_profile_test.cc54 HloCostAnalysis cost_analysis(shape_size_function); in TEST_F() local
57 CreateHloProfilePrinterData(profile_index_map, cost_analysis, in TEST_F()
Dhlo_execution_profile.h100 const HloCostAnalysis& cost_analysis, const string& entry_computation_name);
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dparallel_task_assignment.cc55 std::unique_ptr<HloCostAnalysis> cost_analysis) in DefaultCostModel() argument
58 cost_analysis_(std::move(cost_analysis)) {} in DefaultCostModel()
114 auto cost_analysis = absl::make_unique<HloCostAnalysis>(shape_size); in ParallelTaskAssignment() local
116 Status status = computation->root_instruction()->Accept(cost_analysis.get()); in ParallelTaskAssignment()
120 std::move(cost_analysis))); in ParallelTaskAssignment()
Dcpu_compiler.cc512 HloCostAnalysis cost_analysis(shape_size_bytes); in CreateHloProfilingArtifacts() local
513 TF_RETURN_IF_ERROR(entry_computation.Accept(&cost_analysis)); in CreateHloProfilingArtifacts()
515 **hlo_profile_index_map, cost_analysis, entry_computation.name()); in CreateHloProfilingArtifacts()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnvptx_compiler.cc778 HloCostAnalysis cost_analysis(ShapeSizeBytesFunction()); in RunBackend() local
779 cost_analysis.set_bytes_per_second( in RunBackend()
781 TF_RETURN_IF_ERROR(module->entry_computation()->Accept(&cost_analysis)); in RunBackend()
784 cost_analysis.bytes_accessed()); in RunBackend()
788 *profile_index_map, cost_analysis, entry_computation->name()); in RunBackend()