Searched refs:cost_analysis (Results 1 – 6 of 6) sorted by relevance
51 const HloCostAnalysis& cost_analysis, in CreateHloProfilePrinterData() argument102 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()
54 HloCostAnalysis cost_analysis(shape_size_function); in TEST_F() local57 CreateHloProfilePrinterData(profile_index_map, cost_analysis, in TEST_F()
100 const HloCostAnalysis& cost_analysis, const string& entry_computation_name);
55 std::unique_ptr<HloCostAnalysis> cost_analysis) in DefaultCostModel() argument58 cost_analysis_(std::move(cost_analysis)) {} in DefaultCostModel()114 auto cost_analysis = absl::make_unique<HloCostAnalysis>(shape_size); in ParallelTaskAssignment() local116 Status status = computation->root_instruction()->Accept(cost_analysis.get()); in ParallelTaskAssignment()120 std::move(cost_analysis))); in ParallelTaskAssignment()
512 HloCostAnalysis cost_analysis(shape_size_bytes); in CreateHloProfilingArtifacts() local513 TF_RETURN_IF_ERROR(entry_computation.Accept(&cost_analysis)); in CreateHloProfilingArtifacts()515 **hlo_profile_index_map, cost_analysis, entry_computation.name()); in CreateHloProfilingArtifacts()
778 HloCostAnalysis cost_analysis(ShapeSizeBytesFunction()); in RunBackend() local779 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()